如何為 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

您可以遍歷條形圖並使用set_color()手動更新顏色。

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

您可以在plot中使用color參數

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

為顏色列出這樣的列表並不是很好，但如果你知道情節中總是有 20 個俱樂部，那么它就可以了