基于指定索引值筛选多列数据并将范围外值置零的Python实现

你的代码无法得到正确结果，主要有3个问题：

• 循环写法错误：for i,data in enumerate(data)会覆盖原始存储全量数据的data变量，循环结束后data仅保留最后一行的切片结果，完全没有实现按列处理的逻辑
• 没有置0逻辑：直接对数组切片只会截取部分数据，不会自动将范围外的数值设为0
• 硬编码了第一列的索引规则，没有遍历所有列处理，无法适配多列场景

核心思路是先创建和原数据尺寸一致的全0数组，再逐列把需要保留的行数据复制到对应位置即可，自动适配任意列数的场景，还做了边界保护避免索引越界：

import numpy as np

# 读入原始数据
data = np.loadtxt("inp.dat")
# 每列对应的中心行索引（0基）
index_value = [3,4,4]

# 初始化和原数据同尺寸的全0结果数组，范围外的值默认就是0
res = np.zeros_like(data)

# 逐列处理
for col_idx, center in enumerate(index_value):
    # 计算保留行的起止位置，左闭右开，同时做边界保护防止索引超出数据范围
    start = max(0, center - 2)
    end = min(data.shape[0], center + 3)
    # 把需要保留的行数据复制到结果数组对应位置
    res[start:end, col_idx] = data[start:end, col_idx]

# 保存结果，fmt参数控制输出保留1位小数，和预期格式一致
np.savetxt("file.out", res, fmt="%.1f")

运行上述代码后生成的file.out内容和预期结果完全一致：

0.0 0.0 0.0
2.0 0.0 0.0
3.0 4.0 8.0
4.0 1.0 7.0
5.0 2.0 8.0
5.0 2.0 8.0
0.0 2.0 5.0
0.0 0.0 0.0

后续需要处理更多列时，只需要按列的顺序在index_value列表里追加对应列的中心行索引即可，不需要修改其他逻辑。如果需要调整保留的前后范围，只需要修改center - 2和center + 3里的数字即可，比如要保留前后各3行就改成center-3和center+4。

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