使用plt.plot使用legend

使用plt.plot绘制图形时，可以使用plt.legend()函数添加图例。下面是一个例子：

import matplotlib.pyplot as plt

# 绘制两条曲线
x = [1, 2, 3, 4, 5]
y1 = [1, 4, 9, 16, 25]
y2 = [1, 2, 4, 8, 16]
plt.plot(x, y1, label='y=x^2')
plt.plot(x, y2, label='y=2^x')

# 添加图例
plt.legend()

# 显示图形
plt.show()

在上面的例子中，我们首先使用plt.plot()函数绘制了两条曲线，然后使用plt.legend()函数添加了图例。plt.legend()函数会自动将每条曲线的label属性作为图例的标签。如果没有设置label属性，则可以在plt.legend()函数中手动指定标签。