Matplotlib可以自动或手动显示图例,并为其提供绘图句柄。但不知何故,后者在我身上并不能正常工作。举个例子:
legend_handles = {}
lgh, = plt.plot([0, 1], [0, 1], '-r')
lgh, = plt.plot([0, 1], [1, 1], '-r')
legend_handles["a"] = lgh
lgh, = plt.plot([0, 1], [1, 0], '-b')
legend_handles["b"] = lgh
plt.legend(legend_handles);
这将产生一个带有两条红线的图例,而不是一条蓝色和一条红线。
如何才能使图例仅显示给所选的绘图?
发布于 2019-01-27 17:06:02
图例支持将字典作为输入的no indication。相反,签名是以下任一项
legend() ## (1)
legend(labels) ## (2)
legend(handles, labels) ## (3)
这里使用的是(2),因此在这三行中,只有前两行用字典的键进行了标记(因为字典只有两个键)。
如果您需要使用字典,您需要首先将其解包,以获得两个列表,这两个列表可用于实现情况(3)。
import matplotlib.pyplot as plt
legend_handles = {}
lgh1, = plt.plot([0, 1], [0, 1], '-r')
lgh2, = plt.plot([0, 1], [1, 1], '-r')
legend_handles["a"] = lgh1
lgh3, = plt.plot([1, 0], [1, 0], '-b')
legend_handles["b"] = lgh3
labels, handles = zip(*legend_handles.items())
plt.legend(handles, labels)
plt.show()
然而,完全不使用字典似乎更简单:
import matplotlib.pyplot as plt
lgh1, = plt.plot([0, 1], [0, 1], '-r')
lgh2, = plt.plot([0, 1], [1, 1], '-r')
lgh3, = plt.plot([1, 0], [1, 0], '-b')
plt.legend([lgh1, lgh3], list("ab"))
plt.show()
不要忘记,通过将label
直接提供给艺术家来创建传奇的规范解决方案,
import matplotlib.pyplot as plt
lgh1, = plt.plot([0, 1], [0, 1], '-r', label="a")
lgh2, = plt.plot([0, 1], [1, 1], '-r')
lgh3, = plt.plot([1, 0], [1, 0], '-b', label="b")
plt.legend()
plt.show()
所有情况下的结果:
https://stackoverflow.com/questions/54390421
复制相似问题