使用每个类别的行数对海生线图例进行注释

Question

我想在海生线图中获取并显示每种类型的行数。例如import seaborn as snsfmri = sns.load_dataset("fmri")ax = sns.lineplot(x="timepoint", y="signal", hue="event", data=fmri)我想显示带有事件“stim”的行数和带有事件“cue”的行数作为对图例的补充，例如，它可以显示“stim（23）”，而不是显示图例中的“stim”，这意味着23行将事件作为“stim”

Answer 1

像这样的东西可以解决问题：

fmri = sns.load_dataset("fmri")

x_col = 'timepoint'

y_col = 'signal'

hue_col = 'event'

ax = sns.lineplot(x=x_col, y=y_col, hue=hue_col, data=fmri)

handles,labels = ax.get_legend_handles_labels()

counts = fmri[hue_col].value_counts()

# labels[0] is used for the title by seaborn

new_labels = [labels[0]]+['{:s} ({:d})'.format(l, counts[l]) for l in labels[1:]]

ax.legend(handles, new_labels)