matplotlib.figure中添加图例的方法介绍
在绘图时,为了清晰地表示每个线条或点的含义,通常需要在图中添加图例。在Matplotlib中,可以使用matplotlib.pyplot中的legend()函数或者在matplotlib.figure.Figure对象中使用legend()方法来添加图例。
使用matplotlib.pyplot.legend()函数添加图例:
1. 首先绘制需要添加图例的线条或点。
2. 然后调用matplotlib.pyplot.legend()函数,可以通过设置不同的参数来指定图例的位置、边框样式、字体大小等。
import matplotlib.pyplot as plt import numpy as np x = np.linspace(0, 10, 100) y1 = np.sin(x) y2 = np.cos(x) plt.plot(x, y1, label='sin') plt.plot(x, y2, label='cos') plt.legend(loc='upper right', frameon=False, fontsize=12) plt.show()
在上面的例子中,首先绘制了两条曲线,然后使用label参数给每条曲线起了一个名字。然后调用plt.legend()函数,在参数中指定图例的位置为右上角('upper right'),边框样式为无边框(frameon=False),字体大小为12。调用plt.show()函数显示图像。
使用Figure对象的legend()方法添加图例:
1. 首先绘制需要添加图例的线条或点。
2. 然后获取当前的Figure对象,使用Figure对象的legend()方法,可以通过设置不同的参数来指定图例的位置、边框样式、字体大小等。
import matplotlib.pyplot as plt import numpy as np x = np.linspace(0, 10, 100) y1 = np.sin(x) y2 = np.cos(x) fig, ax = plt.subplots() ax.plot(x, y1, label='sin') ax.plot(x, y2, label='cos') ax.legend(loc='upper right', frameon=False, fontsize=12) plt.show()
在上面的例子中,首先绘制了两条曲线,然后通过plt.subplots()函数获取当前的Figure对象和Axes对象。然后调用ax.legend()方法,在参数中指定图例的位置为右上角('upper right'),边框样式为无边框(frameon=False),字体大小为12。调用plt.show()函数显示图像。
在上述两个例子中,可以通过修改legend()函数或方法的参数来实现更多功能,如调整图例位置('upper right'、'upper left'等)、设置图例的标题(title参数)、设置图例的透明度(alpha参数)等等。具体的参数设置可以参考Matplotlib官方文档。
