使用for循环在matplotlib中创建多个子图

在matplotlib中，可以使用for循环来创建多个子图（subplots）。这种方法对于绘制多个具有相似结构或数据的图表非常有用。下面是一个简单的示例代码，演示如何使用for循环在matplotlib中创建多个子图。
首先，确保已经安装了matplotlib库。如果尚未安装，可以使用以下命令进行安装：

pip install matplotlib

接下来，导入所需的库和模块：

import matplotlib.pyplot as plt
import numpy as np

现在，我们将创建一个简单的数据集，用于在子图中进行绘制。在这个例子中，我们将生成一些随机数据：

# 生成随机数据
data = np.random.rand(10, 5)

现在，我们将使用for循环来创建多个子图。在此示例中，我们将创建3个子图，每个子图包含2行和2列。以下是创建子图的代码：

# 创建子图网格
fig, axs = plt.subplots(2, 2)
# 使用for循环遍历子图并绘制数据
for i in range(2):
for j in range(2):
axs[i, j].imshow(data[i * 2 + j], cmap='viridis')
axs[i, j].set_title(f'Subplot {i}, {j}')
axs[i, j].axis('off')  # 关闭坐标轴标签和刻度

在这个例子中，我们使用了imshow()函数来绘制数据。imshow()函数将数据绘制为图像，其中cmap='viridis'指定了颜色映射。我们还将每个子图的标题设置为Subplot {i}, {j}，其中i和j是当前子图的索引。最后，我们使用axis('off')关闭坐标轴标签和刻度。
现在，我们可以显示绘制的子图：

plt.show()

这将显示一个包含多个子图的图形窗口。每个子图都显示了随机生成的数据。通过使用for循环，我们可以轻松地创建多个具有相似结构的子图，从而更高效地绘制多个图表。