我有以下Python代码。我无法获得图表中条形的颜色以匹配x轴上的颜色名称。我想要" red"有红色条和"白色"有蓝色条形图。但是输出似乎产生了两者的随机组合。
import pandas as pd
import matplotlib.pyplot as plt
colors={'red','blue'}
counts = df.groupby(['quality', 'color']).count()
counts.plot(kind='bar', title="Counts by Wine Quality and Color", color=colors, alpha=.7)
print(counts)
plt.ylabel('Count', fontsize=18)
plt.xlabel('Quality and Color', fontsize=18)
plt.show()
以下是数据集文本:
quality color
3 red 10
white 20
4 red 53
white 163
5 red 681
white 1457
6 red 638
white 2198
7 red 199
white 880
8 red 18
white 175
9 white 5
此代码使用Python 3.6运行: