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

che*_*ens 2 python matplotlib bar-chart dataframe pandas

我有以下数据框:

                        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
Run Code Online (Sandbox Code Playgroud)

我按first_day_weekvar列分组,我需要绘制一个条形图,其中我first_day_week在x轴上,对于first_day_week两个条形图中的每个条目,每个值var用不同的颜色表示,类似于(以下条形图的数据完全是假的):

在此处输入图片说明

jez*_*ael 6

您需要重塑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)
Run Code Online (Sandbox Code Playgroud)

图形

对于排序索引,需要将其转换为日期时间,然后进行排序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)
Run Code Online (Sandbox Code Playgroud)

图1