pythonmatplotlib绘图图例

在Python中使用matplotlib绘图时，图例是非常重要的一部分，它可以帮助我们更好地理解图表中的数据。下面是两种常见的绘制图例的方法：

1.使用label参数和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='y1')
plt.plot(x, y2, label='y2')

# 添加图例
plt.legend()

# 显示图表
plt.show()

2.使用handles和labels参数绘制图例

import matplotlib.pyplot as plt
from matplotlib.lines import Line2D

 生成数据
x = [1, 2, 3, 4, 5]
y1 = [1, 4, 9, 16, 25]
y2 = [1, 2, 3, 4, 5]

# 绘制折线图
line1 = Line2D(x, y1, color='blue', label='y1')
line2 = Line2D(x, y2, color='red', label='y2')
plt.gca().add_artist(line1)
plt.gca().add_artist(line2)

# 添加图例
plt.legend(handles=[line1, line2])

# 显示图表
plt.show()