matplotlib图例无法正确使用句柄

czq61nw1  于 2023-10-24  发布在  其他
关注(0)|答案(1)|浏览(80)

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);

这将给予一个带有两条红线的图例,而不是一条蓝线和一条红线。

我如何让它只显示一个图例到一个选择的情节?

bcs8qyzn

bcs8qyzn1#

存在no indication,图例将支持字典作为输入。

legend()                  ## (1)
legend(labels)            ## (2)
legend(handles, labels)   ## (3)

这里使用的是(2),所以在这三行中,只有前两行用字典的键标记(因为字典只有两个键)。
如果你需要使用字典,你需要先解包它,以获得两个可以用来实现case(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()

所有情况下的结果:

相关问题