使用Matplotlib绘制图形时,可以通过调用plt.legend()方法来设置图例。该方法通常在绘制完所有需要显示在图例中的图形后调用。
例如,可以在绘制完折线图后调用plt.legend()方法来设置图例:
import matplotlib.pyplot as plt
x = [1, 2, 3, 4, 5]
y1 = [1, 4, 9, 16, 25]
y2 = [1, 2, 3, 4, 5]
plt.plot(x, y1, label='Line 1')
plt.plot(x, y2, label='Line 2')
plt.legend()
plt.show()
在上面的例子中,我们使用label参数为每条线设置名称,然后调用plt.legend()方法显示图例。plt.legend()方法可以接受一些参数来调整图例的位置、边框等属性,例如loc参数可以设置图例位置,bbox_to_anchor参数可以设置图例相对于图形的位置等。
除了使用plt.legend()方法外,也可以直接在绘制图形时传入label参数来设置图例,然后调用plt.legend()方法显示图例。