如何删除matplotlib图例中的额外字符串"Line2D"

Question

我在matplotlib图例的开头附加了一个字符串"Line2D".如何摆脱它？重现此问题的简单python脚本如下:

import numpy as np
import matplotlib.pylab as plt

x=np.linspace(0,1,20)
y=np.sin(2*x)
z=np.cos(2*x)

p1, = plt.plot(x,y, label='sin(x)')
p2, = plt.plot(x,z, label='cos(x)')
plt.legend([p1, p2])
plt.show()

我得到一个数字,我想摆脱传奇中的额外字符串"Line2D".我没有足够的声誉来发布图片.如果重要的话,我正在使用anaconda python.谢谢你的帮助!

Answer 1

如果只传递一个列表legend,则必须是要显示的标签,而不是要显示其标签的对象.它将这些行对象转换为字符串,这给出了Line2D(...).

由于您在创建行时标记了行标签,因此无需向其传递任何内容legend.只需plt.legend()将自动使用你提供的标签.

Answer 2

您可以使用plt.legend(handles=[p1, p2]).