没有传奇 Seaborn 线图

Question

当我使用 Seaborn 在同一轴上制作具有多个线图的线图时，不会创建任何图例。 即使我在 sns.lineplot 中为图例提供参数“brief”或“full”，也没有任何显示，并且调用 ax.get_legend_handles_labels() 返回两个空列表。

如何在右侧的框中添加图例，将线条的颜色与名称相关联？

import seaborn as sns
import matplotlib.pyplot as plt
import random

fig1 = plt.figure(figsize=(12, 12))
ax = fig1.add_subplot(1, 1, 1)

x = range(10)

series = list()
for i in range(3):
    y_i = list()
    for j in range(10):
        y_i.append(random.randint(0, 50))

    sns.lineplot(x=x, y=y_i, ax=ax, legend='brief')

plt.show()

Answer 1

那是因为您将简单列表应用为 x 和 y 数据。 它们没有名称或标签等元数据。 您可以通过label -kwarg 添加标签，例如：

sns.lineplot(x=x, y=y_i, ax=ax, legend='brief', label=str(i))

或者您决定为您的数据使用 pandas 系列对象或数据框，它们能够提供列名。

Answer 2

您还可以在创建绘图后在单独的调用中添加图例：

ax.legend(['label 1', 'label 2', 'label 3'])

它在您使用二维数组而不是循环中的多个值列表创建线图的情况下很有用。 在您的情况下，这可能如下所示：

data = np.random.randint(0, 50, size=(10, 3))  # results in 3 lines of length 10 each
sns.lineplot(data=data, ax=ax)