python - Seaborn 不使用 %matplotlib 笔记本显示图例

Question

当我使用%matplotlib笔记本时，图例不显示。

%matplotlib notebook
...
fg = seaborn.FacetGrid(data=df, hue='Scenario', hue_order=df.Scenario.unique().tolist(), aspect=1.61)
fg.map(plt.scatter, 'Yaw', 'posNoseEye').add_legend()

当我使用 %matplotlib inline 时，图例会正确显示。

%matplotlib inline
...
fg = seaborn.FacetGrid(data=df, hue='Scenario', hue_order=df.Scenario.unique().tolist(), aspect=1.61)
fg.map(plt.scatter, 'Yaw', 'posNoseEye').add_legend()

如何在 %matplotlib 笔记本中管理它？ 谢谢

由于我无法在评论中添加图片，所以我尝试了下面的答案，但仍然不起作用。 我有最新版本的 matplotlib 和 seaborn。

Answer 1

该图没有任何用于图例的空间。有多种方法可以克服这个问题，如 this answer 所示。 。

一个选项是手动为图例创建一些空间，例如

%matplotlib notebook
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
import seaborn as sns

df = pd.DataFrame(np.random.rand(20,2), columns=['Yaw', 'posNoseEye'])
df["Scenario"] = np.random.choice(["Scenario A","Scenario B"], size=20)

fg = sns.FacetGrid(data=df, hue='Scenario', hue_order=df.Scenario.unique().tolist(), aspect=1.61)
fg.map(plt.scatter, 'Yaw', 'posNoseEye')

fg.fig.legend()

fg.fig.subplots_adjust(right=0.7)
plt.show()