Question

我有以下数据框：

                        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和var列分组，我需要制作一个条形图，其中我在x轴上有first_day_week，在first_day_week中每个条目都有两个条形图var中不同颜色的值，类似于（以下条形图的数据完全假）：

Answer 1

您需要按unstack重新塑造，然后按DataFrame.plot.bar绘制：

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)

对于排序索引需要转换为datetime，排序然后reindex：

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)

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

1 个答案: