使用 matplotlib 和 pandas 制作 groupby 图

Posted 2023-03-11

【中文标题】使用 matplotlib 和 pandas 制作 groupby 图

【英文标题】：making groupby plot using matplotlib and pandas

【发布时间】：2021-12-27 12:10:58

【问题描述】：

我有一个代码将图形绘制为 fllows

代码是

import matplotlib.pyplot as plt
import pandas as pd

years=["ASD","MNG","KQR","MND","QST", "MNR"]
dataavail=
    "Jan":[20,20,30,19,10,21],
    "Feb":[20,13,10,18,15,30],
    "Mar":[20,20,10,15,18,30],
    "Apr":[20,20,10,15,18,0],
    "May":[20,20,10,15,18,0],
    "Jun":[20,20,10,15,18,0],
    "Jul":[20,20,10,15,18,0],
    "Aug":[20,20,10,15,18,45],
    "Sep":[20,20,10,15,18,0],
    "Oct":[20,20,10,15,18,0],
    "Nov":[20,20,10,15,18,0],
    "Dec":[20,20,0,0,0,0],



df_month = pd.DataFrame(dataavail, index=years)
fig, ax1 = plt.subplots(1, figsize=(8, 5))
df_month.plot(kind='bar', stacked=True, colormap=plt.cm.tab20, ax=ax1)
plt.legend(loc="upper right", ncol = 3,handlelength=1.5, borderpad=0.2, labelspacing=0.2) 
plt.xticks(rotation=0)
plt.show()

但是，我需要 groupby 绘图而不是堆栈图，如图所示，其中 x 轴应与第一个图相同。有人可以帮我解决这个问题。

【问题讨论】：

你要展示多少年？

仅 2016,2017,2018,2019,2020,2021

那么@mozway 的回答有什么问题？

如果我在 x 轴上有很多元素，例如 ASD、MNG... 假设接近 50，那么这些条形将无法正常显示

【参考方案1】：

IIUC 你想要一个未堆叠的条形图吗？

df_month.plot.bar()

输出：

使用您的代码：

您应该删除stacked=True（或使用stacked=False）：

df_month = pd.DataFrame(dataavail, index=years)
fig, ax1 = plt.subplots(1, figsize=(8, 5))
df_month.plot(kind='bar', stacked=False, colormap=plt.cm.tab20, ax=ax1)
plt.legend(loc="upper right", ncol = 3,handlelength=1.5, borderpad=0.2, labelspacing=0.2) 
plt.xticks(rotation=0)
plt.show()

【讨论】：

好的，如果我在 x 轴上有很多元素，比如 ASD、MNG...假设接近 50，那么这些图就无法正常显示，如何克服

使用barh而不是bar，你会得到可以容忍更多值的水平条

感谢@mozway，它帮助了我