Numpy数组如何将指定索引范围外的所有元素值替换为0

核心逻辑很简单：要么先生成同形状全0数组，把需要保留的索引位对应值填进去；要么直接修改原数组，把不需要保留的位置赋值为0。

已知条件 #

• 原始数组元素：[1, 2, 3, 4, 5]
• 目标效果：仅保留索引为2（NumPy索引从0开始计数，对应值为3的位置）的元素，其余位置替换为0，最终输出[0, 0, 3, 0, 0]
• 原有代码注意点：Import首字母大写不符合Python语法规范，需要改成小写import

可直接运行的实现代码 #

方法1：生成全0副本填充目标值（不会修改原始数组数据） #

import numpy as np
data = np.loadtxt('data.txt')
# 生成和原始数组形状、数据类型完全一致的全0数组
res = np.zeros_like(data)
# 配置需要保留的索引起止范围，单元素保留时起止索引设为同一个值即可
keep_idx_start = 2
keep_idx_end = 2
# 把保留范围内的原始值填充到全0数组对应位置
res[keep_idx_start : keep_idx_end + 1] = data[keep_idx_start : keep_idx_end + 1]
print(res)

方法2：直接修改原数组（会覆盖原始数组数据） #

import numpy as np
data = np.loadtxt('data.txt')
# 保留范围之前的元素全部置0
data[:keep_idx_start] = 0
# 保留范围之后的元素全部置0
data[keep_idx_end + 1:] = 0
print(data)

扩展说明 #

• 如果需要保留一段连续索引，比如保留索引1到3的元素，只需要把keep_idx_start设为1、keep_idx_end设为3即可
• 如果需要保留不连续的索引位置，可以直接传入索引列表，比如要保留索引0和索引3的元素，写法为：

keep_indices = [0, 3]
res = np.zeros_like(data)
res[keep_indices] = data[keep_indices]

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