使用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()方法顯示圖例。