python - Pandas 绘图条顺序类别

Question

我有一个包含三个唯一值“低”、“中”和“高”的分类变量的数据集:

df.CatVar.value_counts()
Out[93]: 
Medium    35832
Low       25311
High      12527
Name: CatVar, dtype: int64

我正在尝试将唯一值的数量绘制为条形图。但是，以下代码按 ["Medium"、"Low"、"High"] 的顺序给我条形图

df.CatVar.value_counts().plot(kind="bar")

如何更改图中条形的顺序？

Answer 1

有 2 种可能的解决方案 - 在 plot 之前更改 index 的顺序 - by reindex或 loc:

df.CatVar.value_counts().reindex(["Low", "Medium", "High"]).plot(kind="bar")

df.CatVar.value_counts().loc[["Low", "Medium", "High"]].plot(kind="bar")

或者使用ordered categorical ，所以在 value_counts 之后通过 categories 参数获得顺序:

df.CatVar = pd.Categorical(df.CatVar, categories=["Low", "Medium", "High"], ordered=True)
df.CatVar.value_counts(sort=False).plot(kind="bar")

---

示例:

df = pd.DataFrame({'CatVar':['Low','Medium','Low','Low','Medium','High']})
print (df)
   CatVar
0     Low
1  Medium
2     Low
3     Low
4  Medium
5    High

df.CatVar.value_counts().reindex(["Low", "Medium", "High"]).plot(kind="bar")