如何在Matplotlib中为已绘制的线设置标签?
当我们绘制一张图时,有时候需要为图中的线条设置标签,方便更好地展示和阐述数据。而在Matplotlib中,为已绘制的线设置标签也是非常简单的。
set_label
在Matplotlib中,我们可以使用set_label()方法为已绘制的线条设置标签。下面我们来看一个简单的例子,首先生成一些随机数据:
import numpy as np
import matplotlib.pyplot as plt
x = np.linspace(0, 2*np.pi, 200)
y1 = np.sin(x)
y2 = np.cos(x)
然后我们绘制这两条线:
plt.plot(x, y1)
plt.plot(x, y2)
现在我们为这两条线设置标签:
plt.plot(x, y1)
plt.plot(x, y2)
plt.plot([], [], label='sin') # 空的线条,只为设置标签
plt.plot([], [], label='cos') # 空的线条,只为设置标签
plt.legend()
plt.show()
在代码中,我们使用了两个空的线条来设置标签,这两条线条没有实际的数据,只是为了说明标签的用途。接下来使用legend()方法显示标签。
在图中,我们可以看到每条线都有了对应的标签。
完整代码
下面是完整的示例代码:
import numpy as np
import matplotlib.pyplot as plt
x = np.linspace(0, 2*np.pi, 200)
y1 = np.sin(x)
y2 = np.cos(x)
plt.plot(x, y1)
plt.plot(x, y2)
plt.plot([], [], label='sin') # 空的线条,只为设置标签
plt.plot([], [], label='cos') # 空的线条,只为设置标签
plt.legend()
plt.show()
结论
在Matplotlib中为已绘制的线设置标签非常简单,只需要使用set_label()方法,然后使用legend()方法显示标签即可。