将观察数量添加到水平箱线图 seaborn

Add number of observations to horizontal boxplot seaborn

我想在 seaborn 的水平箱线图中添加观察值 (nobs) 的数量 - 最好在晶须上方,如果不可能,则在中值旁边。这些位置中的任何一个的替代方案也可以,但最不可取。

为此,我尝试改编其他地方的代码 (https://python-graph-gallery.com/38-show-number-of-observation-on-boxplot/),展示如何针对垂直箱线图执行此操作,但无法将其应用于水平箱线图。

这是我目前的尝试:

df = sns.load_dataset("iris")

ax = sns.boxplot(y="species", x="sepal_length", data=df)
 
# Calculate number of obs per group & median to position labels
medians = df.groupby(['species'])['sepal_length'].median().values
nobs = df['species'].value_counts().values
nobs = [str(x) for x in nobs.tolist()]
nobs = ["n: " + i for i in nobs]
 
# Add it to the plot
pos = range(len(nobs))
for tick,label in zip(pos,ax.get_xticklabels()):
 ax.text(pos[tick], medians[tick],nobs[tick],
         color='black', verticalalignment = "bottom")

sns.despine()

plt.show();

结果显示了箱线图之外的观察值(nobs)的数量 - 请参见下图的左下角。

如何按照问题顶部所述以编程方式将 (nobs) 移动到图中?

根据@JohanC 的评论,它运行良好。请检查代码段

import matplotlib.pyplot as plt
import seaborn as sns
df = sns.load_dataset("iris")
ax = sns.boxplot(y="species", x="sepal_length", data=df) 
medians = df.groupby(['species'])['sepal_length'].median().values
nobs = df['species'].value_counts().values
nobs = [str(x) for x in nobs.tolist()]
nobs = ["n: " + i for i in nobs] 
pos = range(len(nobs))
for tick,label in zip(pos,ax.get_yticklabels()):
    ax.text(medians[tick],tick,nobs[tick],color='black', verticalalignment = "bottom")
sns.despine()
plt.show();