seaborn - 如何将平均条添加到与色调分组的条形图中

Question

我的代码如下：

import pandas as pd

import numpy as np

import seaborn as sns

data=[np.random.randint(2018,2020,size=(30)),      
      np.random.randint(1,13,size=(30)),
      np.random.randint(1,101,size=(30)),
      np.random.randint(1,101,size=(30))]

cols=['year','month','val','val1']

data=pd.DataFrame(data).T

data.columns=cols

data1=[np.random.randint(1,13,size=(30)),
      np.random.randint(1,101,size=(30)),
      np.random.randint(1,101,size=(30))]

cols1=['month','val','val1']

data1=pd.DataFrame(data1).T

data1.columns=cols1


sns.barplot(data=data,x='month',y='val',hue='year',ci=False)

sns.barplot(data=data,x='month',y='val',estimator=np.mean,ci=False)

产生条形图，实际上我得到了两个条形图

第二个是每个月的平均值

但我想要一个 plot 每个月包含三列，包括平均条。 你能帮我解决这个问题吗？

Answer 1

您可以使用熊猫的 plot function：

(data.pivot_table(index='month',columns='year',
                  values='val', margins=True, 
                  margins_name='Mean')
     .drop('Mean')
     .plot.bar()
)

Output：