不同顏色的列，Padas DataFrame

Question

我有這樣的 Pandas DataFrame：

data = pd.DataFrame({"fuel":["gas","gas","diesel","diesel","gas","diesel"]})

我用代碼來制作條形圖：

ax1 = data.fuel.value_counts().plot('bar')
ax1.set(xlabel = 'Fuel Type', ylabel='Frequency of fuel type')

盡管如此，我的兩列都在一個顏色（藍色）中。 我應該怎么做才能有不同顏色的列？

Answer 1

您可以在 plot() 中設置顏色參數。

import pandas as pd
from matplotlib import pyplot as plt

data = pd.DataFrame({"fuel":["gas","gas","diesel","diesel","gas","diesel"]})

data.fuel.value_counts().plot('bar', color=['black', 'red'])
ax1.set(xlabel = 'Fuel Type', ylabel='Frequency of fuel type')

plt.show()

Answer 2

根據data.fuel的唯一值的大小創建隨機顏色列表，並將其傳遞給plot選項color

colors = [np.random.uniform(size=3) for _ in  range(data.fuel.nunique())]
ax1 = data.fuel.value_counts().plot('bar', color=colors)
ax1.set(xlabel = 'Fuel Type', ylabel='Frequency of fuel type')