![](/img/trans.png)
[英]How to assign different position for each group in seaborn violin plot?
[英]Is it possible to annotate a seaborn violin plot with number of observations in each group?
我想用每组中的观察数量来注释我的小提琴情节。 所以问题基本上与这个问题相同,除了:
让我们从Seaborn API文档中获取此示例:
import seaborn as sns
sns.set_style("whitegrid")
tips = sns.load_dataset("tips")
ax = sns.violinplot(x="day", y="total_bill", data=tips)
我希望在小提琴之上有n = 62,n = 19,n = 87,并且n = 76。 这可行吗?
在这种情况下,我喜欢预先计算带注释的值并将它们合并到分类轴中。 换句话说,预计算例如“Thurs,N = xxx”
看起来像这样:
import seaborn as sns
sns.set_style("whitegrid")
ax= (
sns.load_dataset("tips")
.assign(count=lambda df: df['day'].map(df.groupby(by=['day'])['total_bill'].count()))
.assign(grouper=lambda df: df['day'].astype(str) + '\nN = ' + df['count'].astype(str))
.sort_values(by='day')
.pipe((sns.violinplot, 'data'), x="grouper", y="total_bill")
.set(xlabel='Day of the Week', ylabel='Total Bill (USD)')
)
您首先需要存储y位置和x位置的所有值(使用您的数据集)以便使用ax.text
,然后一个简单的for
循环可以在所需位置写入所有内容:
import seaborn as sns
import matplotlib.pyplot as plt
tips = sns.load_dataset("tips")
ax = sns.violinplot(x="day", y="total_bill", data=tips)
yposlist = tips.groupby(['day'])['total_bill'].median().tolist()
xposlist = range(len(yposlist))
stringlist = ['n = 62','n = 19','n = 87','n = 76']
for i in range(len(stringlist)):
ax.text(xposlist[i], yposlist[i], stringlist[i])
plt.show()
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.