python作图如何添加图例

在Python中，使用matplotlib库做图时，添加图例（Legend）可以帮助读者理解各个线条代表的内容。以下是如何在绘制函数图时添加图例的步骤：

1. 首先，确保你已经导入了matplotlib的`pyplot`模块，并创建了一个绘图窗口：

import matplotlib.pyplot as plt

2. 当你绘制多个具有不同标识（如不同线型、颜色或标记）的线条时，需要为每条线设置一个标签。通常，当你调用`plot()`函数时，可以传递第三个参数作为标签：

# 示例：绘制两条线，一条用蓝色实线，另一条用红色虚线
line1, = plt.plot(x_data, function_1(x_data), 'b-', label='Function 1')
line2, = plt.plot(x_data, function_2(x_data), 'r--', label='Function 2')

# 这里，`line1, = ...` 和 `line2, = ...` 是为了简化返回结果的处理，`=`用于获取`plot()`的返回值，而逗号将它赋值给别名

3. 添加图例的位置，你可以选择左下角、右下角或者其他位置，通过`legend()`函数并指定位置参数来完成：

plt.legend(loc='upper right')  # 可替换为其他位置，如'best', 'center left', etc.

4. 显示图例后，最后显示整个图形：

plt.show()

如果只有一个函数，也可以直接在`plot()`调用后添加标签，然后不需要单独的`legend()`命令。