plt 强制轴数 - Python

在画图时，有时我们需要控制图像的轴数，这在处理多种数据在同一图表上显示时非常有用。在 Python 中，我们可以使用 matplotlib 中的 pyplot 模块的 subplots 函数来实现。

示例

下面的示例展示了如何通过 subplots 函数创建一个具有两个轴的图像，并在其中绘制两个不同的数据集。这里我们使用 pyplot 模块中的 plot 函数来绘制数据。

import matplotlib.pyplot as plt

x1 = [1, 2, 3, 4]
y1 = [10, 20, 30, 40]

x2 = [1, 2, 3, 4]
y2 = [40, 30, 20, 10]

fig, ax = plt.subplots(2, 1)

ax[0].plot(x1, y1)
ax[1].plot(x2, y2)

plt.show()

输出结果:

解释

在上述示例中，我们首先定义两组数据集 x1, y1 和 x2, y2，然后使用 subplots 函数创建了一个具有两个轴的图像。变量 fig 包含整个图像对象，而 ax 是一个轴对象的数组，使用索引访问轴对象。

我们在两个轴对象上使用 plot 函数分别绘制了数据集 x1, y1 和 x2, y2。最后使用 show 函数显示图像。通过 subplots 函数，我们可以自由控制图像中轴对象的数量、位置和排列方式。

自此，我们成功的使用 subplots 函数创建了一个具有两个轴的图像，并在其中绘制两个不同的数据集。