首页 > 解决方案 > 使用 matplotlib 和 pandas 制作 groupby 图

问题描述

我有一个将图形绘制为流的代码在此处输入图像描述

代码是

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 轴应与第一个图相同。任何人都可以帮助我解决这个问题。 在此处输入图像描述

标签: pandasmatplotlibpandas-groupbybar-chart

解决方案


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()

条形图


推荐阅读