求助：如何将文本文件中指定10列数据分别绘制到子图中

你的代码存在三个核心问题，导致无法正确绘制列数据：

• 子图数量设置冗余：你只需要绘制10列数据，但代码创建了5×10=50个子图，完全没必要，还会引发索引越界
• 数据索引错误：data[count]是取第count行的数据，而你需要的是第count列，应该用data[:, count]
• 子图遍历逻辑错误：循环的行/列索引顺序颠倒，且遍历范围超出了10列的数量

修正后的代码 #

import matplotlib.pyplot as plt
import numpy as np

# 加载数据并提取前10列
data = np.loadtxt("input_data.txt")
data = data[:, 0:10]
print(f"数据形状：{data.shape}")  # 应该输出 (2048, 10)

# 设置子图布局：2行5列，刚好放下10个图，可根据需求调整（比如5行2列）
nrows = 2
ncols = 5
fig, axs = plt.subplots(nrows, ncols, figsize=(15, 8))  # 调整画布大小适配布局

# 扁平化子图数组，方便遍历（避免嵌套循环）
axs_flat = axs.flatten()

for idx in range(data.shape[1]):  # 遍历10列
    # 绘制第idx列的所有行数据
    axs_flat[idx].plot(data[:, idx])
    # 给每个子图加标题，标注列号
    axs_flat[idx].set_title(f"第{idx+1}列数据")
    # 可选：调整坐标轴标签
    axs_flat[idx].set_xlabel("行索引")
    axs_flat[idx].set_ylabel("数值")

# 调整子图间距，避免标题重叠
plt.tight_layout()
plt.show()

关键修正点说明 #

1. 子图布局优化：用2行5列或5行2列的布局，刚好匹配10列数据的需求，避免创建多余子图
2. 数据索引修正：data[:, idx]表示取所有行的第idx列，这才是你需要绘制的单列数据
3. 扁平化子图数组：通过axs.flatten()把二维的子图数组转成一维，直接遍历索引即可，无需嵌套循环，逻辑更清晰
4. 美化细节：添加子图标题、坐标轴标签，用plt.tight_layout()自动调整子图间距，避免元素重叠

如果想要5行2列的布局，只需要把nrows=2改成nrows=5，ncols=5改成ncols=2即可，代码其他部分无需改动。

内容的提问来源于stack exchange，提问作者user19520518