2017-08-03 60 views
3

除了发布在this link上的解决方案之外,我还希望我也可以添加色相参数,并在每个图中添加中间值。带中值的标签盒图

当前代码:

testPlot = sns.boxplot(x='Pclass', y='Age', hue='Sex', data=trainData) 
m1 = trainData.groupby(['Pclass', 'Sex'])['Age'].median().values 
mL1 = [str(np.round(s, 2)) for s in m1] 
p1 = range(len(m1)) 

for tick, label in zip(p1, testPlot.get_xticklabels()): 
    print(testPlot.text(p1[tick], m1[tick] + 1, mL1[tick])) 

给出了一个输出像:enter image description here

我工作泰坦尼克数据集可以在this link找到。

我得到所需的值,但只有当我做一个打印语句时,如何将它包含在我的剧情?

回答

3

将您的标签手动根据所有xticklabels的周期色调参数,并为每个类别栏宽度:

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

trainData = pd.read_csv('titanic.csv') 
testPlot = sns.boxplot(x='pclass', y='age', hue='sex', data=trainData) 
m1 = trainData.groupby(['pclass', 'sex'])['age'].median().values 
mL1 = [str(np.round(s, 2)) for s in m1] 

ind = 0 
for tick in range(len(testPlot.get_xticklabels())): 
    testPlot.text(tick-.2, m1[ind+1]+1, mL1[ind+1], horizontalalignment='center', color='w', weight='semibold') 
    testPlot.text(tick+.2, m1[ind]+1, mL1[ind], horizontalalignment='center', color='w', weight='semibold') 
    ind += 2  
plt.show() 

enter image description here