在子图中绘制均值 - Seaborn

时间:2021-04-29 22:34:12

标签: python pandas seaborn

我的目标是绘制一组子图的平均值。使用下面,我将每个独特的 Item 分成单独的子图。我希望将 Num 的相关均值绘制到每个子图中。

import pandas as pd
import seaborn as sns

df = pd.DataFrame({      
    'Num' : [1,2,1,2,3,2,1,3,2,2,1,2,3,3,1,3],
    'Label' : ['A','B','C','B','B','C','C','B','B','A','C','A','B','A','C','A'],  
    'Item' : ['Up','Left','Up','Left','Down','Right','Up','Down','Right','Down','Right','Up','Up','Right','Down','Left'],        
   })

g = sns.displot(data = df, 
               x = 'Num', 
               row = 'Item', 
               row_order = ['Up','Down','Left','Right'],
               discrete = True,
               aspect = 4, 
               height = 2,
               )

#for x in df['IMMEDIATE_CONGESTION_PLAYER_COUNT']:

plt.axvline(np.median(df['Num']),color='b', linestyle='--')

1 个答案:

答案 0 :(得分:1)

这是一个 great answer。对于初始图形对象,使用map函数添加一条垂直线。

import pandas as pd
import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt

df = pd.DataFrame({      
    'Num' : [1,2,1,2,3,2,1,3,2,2,1,2,3,3,1,3],
    'Label' : ['A','B','C','B','B','C','C','B','B','A','C','A','B','A','C','A'],  
    'Item' : ['Up','Left','Up','Left','Down','Right','Up','Down','Right','Down','Right','Up','Up','Right','Down','Left'],        
   })

g = sns.displot(data = df, 
               x = 'Num', 
               row = 'Item', 
               row_order = ['Up','Down','Left','Right'],
               discrete = True,
               aspect = 4, 
               height = 2,
               )

cols = df['Item'].unique()
for c,ax in zip(cols, g.axes.flat):
    ax.axvline(np.median(df[df['Item'] == c]['Num']), color='r', linestyle='--')

plt.show()

enter image description here