如何为Matplotlib条形图中的特定列设置不同的颜色？

Question

我创建了一个10个足球俱乐部的酒吧图，其中最高和10的FIFA收视率。 我以这种方式绘制了同一数字：

grouped_top=df.groupby("club_name")["overall"].mean().reset_index().sort_values("overall",ascending=False).head(10)
grouped_bottom=df.groupby("club_name")["overall"].mean().reset_index().sort_values("overall",ascending=False).tail(10)
unioned=pd.concat([grouped_top,grouped_bottom])
unioned.plot(kind="bar",x="club_name",y="overall",xlabel="",figsize=(15,5))
plt.show()

结果是： ，

但我想进入前10列和底部10列以具有不同的颜色。例如，前10列是红色的，最后10列蓝色。我也希望我的图表具有一个传奇，可以解释哪种颜色对应于哪种颜色。 我希望有一种方法。

Answer 1

您可以迭代栏上并使用手动更新颜色。

import matplotlib
import matplotlib.pyplot as plt


ax = unioned.plot(kind="bar", x="club_name", y="overall", xlabel="", figsize=(15,5))

# Highest ratings
for i in range(0, 10):
    ax.get_children()[i].set_color("red")

# Lowest ratings
for i in range(10, 20):
    ax.get_children()[i].set_color("blue")

legends = [
    matplotlib.patches.Patch(color="red", label="Highest ratings"),
    matplotlib.patches.Patch(color="blue", label="Lowest ratings"),
]
ax.legend(handles=legends, prop={"size": 20})

plt.show()

Answer 2

您可以在绘图中使用 color 参数

unioned.plot(kind="bar",x="club_name",y="overall",xlabel="",figsize=(15,5), color=[*['red']*10, *['green']*10])

对颜色的列表不太好，但是如果您知道图中总是有20个俱乐部，那么它将执行诡计