Python 画图添加图例教程

1. 整体流程

为了帮助小白开发者实现“python 画图添加图例”的操作，下面是整个过程的步骤流程表格：

步骤 描述
步骤 1 导入所需的库
步骤 2 准备数据
步骤 3 创建画布和子图
步骤 4 绘制图形
步骤 5 添加图例
步骤 6 显示图形

接下来，我将详细讲解每一步需要做什么以及涉及到的代码。

2. 步骤与代码

步骤 1：导入所需的库

首先，我们需要导入一些Python的绘图库，比如Matplotlib和Numpy。可以使用下面的代码导入这些库：

import matplotlib.pyplot as plt
import numpy as np

步骤 2：准备数据

在绘制图形之前，我们需要准备一些数据。这些数据可以是一组数值或者是两组相关的数据。这里我们以绘制折线图为例，使用Numpy生成一些随机数据：

x = np.linspace(0, 10, 100)
y1 = np.sin(x)
y2 = np.cos(x)

步骤 3：创建画布和子图

在绘制图形之前，我们需要创建一个画布和一个子图。画布是一个空白的绘图区域，而子图是在画布上创建的一个小的图形区域。可以使用下面的代码创建画布和子图：

fig, ax = plt.subplots()

步骤 4：绘制图形

现在我们可以使用子图ax来绘制图形了。这里我们以折线图为例，使用plot()函数绘制两条曲线：

ax.plot(x, y1, label='sin(x)')
ax.plot(x, y2, label='cos(x)')

步骤 5：添加图例

为了添加图例，我们需要使用legend()函数。这个函数将自动根据之前在plot()函数中定义的label参数创建图例。可以使用下面的代码添加图例：

ax.legend()

步骤 6：显示图形

最后，我们需要使用show()函数来显示图形：

plt.show()

3. 示例代码

下面是完整的示例代码：

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(x)')
ax.plot(x, y2, label='cos(x)')

# 添加图例
ax.legend()

# 显示图形
plt.show()

4. 效果展示

下面是使用上述代码绘制的折线图效果展示：

stateDiagram
    [*] --> Start
    Start --> Draw
    Draw --> AddLegend
    AddLegend --> Show
    Show --> [*]

journey
    title 绘制折线图的过程
    section 准备数据
    添加随机数据
    section 创建画布和子图
    创建空白画布和子图
    section 绘制图形
    绘制折线图
    section 添加图例
    使用legend函数添加图例
    section 显示图形
    显示绘制的图形

通过上述步骤和示例代码，你应该已经了解了如何在Python中绘制图形并添加图例了。希望这篇文章对你有所帮助！