在Python中,使用matplotlib库可以很容易地添加图例。以下是一个简单的例子,展示了如何使用`plt.legend()`函数来设置图例:
import matplotlib.pyplot as plt
定义数据
x = [1, 2, 3, 4]
y1 = [10, 20, 30, 40]
y2 = [15, 25, 35, 45]
绘制两条线,并为每条线添加标签
plt.plot(x, y1, label='Line 1')
plt.plot(x, y2, label='Line 2')
添加图例,显示在图形的适当位置
plt.legend()
显示图形
plt.show()
在这个例子中,我们首先导入了`matplotlib.pyplot`模块,并定义了两个数据系列`y1`和`y2`。然后,我们使用`plt.plot()`函数绘制这两条线,并通过`label`参数为它们分别指定了标签。最后,我们调用`plt.legend()`函数来生成图例,并通过`plt.show()`函数显示整个图形。
你还可以通过`plt.legend()`函数的其他参数来定制图例的外观,例如位置、颜色、字体大小等。例如,如果你想将图例放置在图形的顶部,可以使用`loc`参数:
plt.legend(loc='upper center')
你还可以设置图例的字体大小和颜色:
plt.legend(fontsize=12, frameon=True, framealpha=0.9, title='Legend')
这里`fontsize`设置了字体大小,`frameon`设置为`True`来显示图例的边框,`framealpha`设置了边框的透明度,`title`设置了图例的标题。