python - 使用一个“绘图”调用绘制多条曲线时的一个图例条目

Question

我通过使用一个plot调用绘制多条曲线来创建一个网格：

import matplotlib.pyplot as plt
import numpy as np

fig, ax = plt.subplots()

x = np.array([[0,1], [0,1], [0,1]])
y = np.array([[0,0], [1,1], [2,2]])

ax.plot([0,1],[0,2], label='foo', color='b')

ax.plot(x.T, y.T, label='bar', color='k')

ax.legend()

plt.show()

生成的图例具有与曲线一样多的“条形”条目（见下文）。我希望每次plot通话只有一个图例条目（在这种情况下只有一次“条形图”）。

我希望这样我可以有其他绘图命令（例如，绘制'foo'曲线的那个），如果它们有标签，它们的曲线会自动包含在图例中。我特别想避免在构造图例时手动选择句柄，而是使用 matplotlib 的功能通过是/否来处理这个问题，包括在绘图时包含一个标签。我怎样才能做到这一点？

标签： pythonmatplotlib

Answer 1

这是一种可能的解决方案：您可以使用下划线不产生图例条目的事实。因此，将除第一个标签之外的所有标签都设置为"_"抑制那些出现在图例中的标签。

import matplotlib.pyplot as plt
import numpy as np

fig, ax = plt.subplots()

x = np.array([[0,1], [0,1], [0,1]])
y = np.array([[0,0], [1,1], [2,2]])

ax.plot([0,1],[0,2], label='foo', color='b')

lines = ax.plot(x.T, y.T, label='bar', color='k')
plt.setp(lines[1:], label="_")
ax.legend()

plt.show()