不同颜色的列，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')