python - 在按 DataFrame 列着色图时，如何设置每个值的颜色？

Question

在 matplotlib(特别是 Pandas )中，如何将特定颜色映射到用于区分颜色的列的值？

假设我有一个专栏......

>> df["country"]
DE
EN
US
DE

...现在我想从 DataFrame 中绘制值，其中每个国家/地区的颜色都不同。如何确定哪个国家获得哪种颜色？带颜色图？不幸的是，我找不到合适的文档。

我想应用这样的字典:

# pseudo-code
colormapping = {"DE": "blue", ...}
df.plot(colorby="country", colormapping)

编辑:

这是一个示例数据帧。

    outlook   play  temperature country
0     sunny   True           25      DE
1     sunny   True           25      EN
2  overcast   True           19      DE
3      rain  False           21      US
4  overcast  False           33      IT
5      rain  False           27      EN
6      rain  False           22      FR
7  overcast   True           26      FR
8     sunny   True           13      FR
9     sunny   True           16      CH

Answer 1

您可以通过指定色调级别到相应 matplotlib 的字典映射来实现。 palette 中的颜色 categorical plot 的参数使用 seaborn如图所示:

sns.set(style="whitegrid")
sns.swarmplot(x="outlook", y="temperature", hue="country", data=df, size=8,
              palette={'DE':'b', 'EN':'g', 'US':'r','IT':'c', 'FR':'y', 'CH':'k'})