如何在 matplotlib 和 pandas 中绘制由两列分组的数据框

2024-03-17

我有以下数据框:

                        total_gross_profit
  first_day_week    var 
      Feb-06         1  45293.09
                     2  61949.54
      Feb-13         1  44634.72
                     2  34584.15
      Feb-20         1  43796.89
                     2  37308.57
      Feb-27         1  44136.21
                     2  38237.67
      Jan-16         1  74695.91
                     2  75702.02
      Jan-23         1  86101.05
                     2  69518.39
      Jan-30         1  65913.56
                     2  74823.94
      Mar-06         1  34256.47
                     2  31953.00

分组依据first_day_week and var我需要在我有的地方制作一个条形图first_day_week在 x 轴上并且对于每个条目first_day_week每个值的两个条形图var用不同的颜色,类似于(以下条形图的数据完全是假的):


你需要重塑unstack http://pandas.pydata.org/pandas-docs/stable/generated/pandas.DataFrame.unstack.html然后绘制DataFrame.plot.bar http://pandas.pydata.org/pandas-docs/stable/generated/pandas.DataFrame.plot.bar.html:

print (df.unstack()['total_gross_profit'])
var                    1         2
first_day_week                    
Feb-06          45293.09  61949.54
Feb-13          44634.72  34584.15
Feb-20          43796.89  37308.57
Feb-27          44136.21  38237.67
Jan-16          74695.91  75702.02
Jan-23          86101.05  69518.39
Jan-30          65913.56  74823.94
Mar-06          34256.47  31953.00


df.unstack()['total_gross_profit'].plot.bar(rot=0)

对于排序索引需要转换为日期时间,然后排序reindex http://pandas.pydata.org/pandas-docs/stable/generated/pandas.Series.reindex.html:

df1 = df.unstack()['total_gross_profit']
idx = pd.to_datetime(df1.index, format='%b-%d').sort_values().strftime('%b-%d')
print (idx)
['Jan-16' 'Jan-23' 'Jan-30' 'Feb-06' 'Feb-13' 'Feb-20' 'Feb-27' 'Mar-06']

df1.reindex(idx).plot.bar(rot=90)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

如何在 matplotlib 和 pandas 中绘制由两列分组的数据框 的相关文章