我正在从seaborn的数据框中创建一个线图,我想向该图添加一条水平线。效果很好,但是我无法在图例中添加水平线。
这是一个最小的可验证示例:
import seaborn as sns
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
x = np.array([2, 2, 4, 4])
y = np.array([5, 10, 10, 15])
isBool = np.array([True, False, True, False])
data = pd.DataFrame(np.column_stack((x, y, isBool)), columns=["x", "y", "someBoolean"])
print(data)
ax = sns.lineplot(x="x", y="y", hue="someBoolean", data=data)
plt.axhline(y=7, c='red', linestyle='dashed', label="horizontal")
plt.legend(("some name", "some other name", "horizontal"))
plt.show()
这将导致以下绘图:
“某些名称”和“某些其他名称”的图例正确显示,但是“水平”图例只是空白。我尝试仅使用plt.legend()
,但图例由数据集中看似随机的值组成。
有什么想法吗?
答案 0 :(得分:3)
仅使用plt.legend()
即可告诉您正在绘制哪些数据:
您使用someBoolean
作为色相。因此,您实际上是通过对数据应用布尔掩码来创建两行。一行代表的是False(在上面的图例中显示为0),另一行代表的是True(在上面的图例中显示为1)。
要获取图例,您需要设置手柄和标签。您可以使用ax.get_legend_handles_labels()
获取它们的列表。然后,请确保省略第一个没有艺术家的手柄:
ax = sns.lineplot(x="x", y="y", hue="someBoolean", data=data)
plt.axhline(y=7, c='red', linestyle='dashed', label="horizontal")
labels = ["some name", "some other name", "horizontal"]
handles, _ = ax.get_legend_handles_labels()
# Slice list to remove first handle
plt.legend(handles = handles[1:], labels = labels)
这给出了: