python-3.x - 在 Seaborn 绘图上绘图并添加图例

Question

请参阅下面的代码。我用 seaborn 创建了一个情节然后我用 plt.plot 在同一图上绘制了一些其他值，工作正常，但 plt.legend方法显然不是可加的，也就是说，我不能用它来标记图表中的两条线，即 'from seaborn'和 'from plt' ，两者都与正确的线条颜色相关联。

import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
mean, cov = [0, 2], [(1, .5), (.5, 1)]
data_sample, y = np.random.multivariate_normal(mean, cov, size=100).T
data_min = min(data_sample)
data_max = max(data_sample)
plot_points = [(data_min + (i/100) * (data_max-data_min)) for i in range(0, 100)]
sns.set(color_codes=True)
plt.figure()
ax = sns.kdeplot(data_sample, legend='from seaborn')
ax.plot(plot_points, y)
plt.legend(['from plt'])
plt.show()

我该如何解决这个问题？

Answer 1

更改 legend至 label并添加 label至 plot :

ax = sns.kdeplot(data_sample, label='from seaborn')
ax.plot(plot_points, y, label='from plt')
ax.legend()

输出: