Я пытался выяснить, как настроить легенду для полосового сюжета, используя данные с компакт-диска C о подростковой беременности. Моя основная идея состояла в том, чтобы создать пузырьковую версию полосового сюжета basi c в Python, которая прекрасно работает. Я только хочу добавить легенду, обозначающую размер пузырьков для простоты интерпретации.
Вот мой код:
fig = plt.figure(figsize=(12, 10))
ax = sns.stripplot(x="Race", y="Year", size=df["Scale"], color="midnightblue", jitter=True, data=df)
ax.set_ylim(1950, 2025)
ax.set_xlabel("")
ax.set_ylabel("")
ax.invert_yaxis()
handles, labels = ax.get_legend_handles_labels()
legend2 = ax.legend(handles, labels, loc="upper right", title="Birth Rate")
ax.set_title("Birth Rate (per 1,000): 1960-2018", fontsize=20)
sns.despine(trim=True)
, который производит эту цифру,
Bubble Strip-plot
Я был изучая руководство о том, как задать / аннотировать пользовательскую легенду с помощью https://matplotlib.org/3.1.0/gallery/lines_bars_and_markers/scatter_with_legend.html, но я не уверен, что делать дальше, используя стрипплотную функцию Seaborn.
Есть идеи?
Best,
Antonio