如何在Numpy数组指定第三轴索引中按条件替换元素？

你的代码报错核心原因是：arr[:,:,indices]提取出的子数组形状为(10,200,5)，由此生成的布尔数组形状也和子数组一致，但原数组形状是(10,200,50)，两者形状不匹配，无法直接用于索引原数组。

下面提供两种可行的解决方案：

方案一：创建全数组布尔掩码 #

先生成一个和原数组形状完全一致的布尔掩码，仅标记需要修改的位置：

import numpy as np

# 示例初始化数组（替换成你的实际数组）
arr = np.random.rand(10, 200, 50) * 100
indices = [0, 1, 15, 20, 19]

# 创建全False的掩码，形状与原数组一致
mask = np.zeros(arr.shape, dtype=bool)
# 在第三轴的指定索引位置，标记数值大于33的元素
mask[:, :, indices] = arr[:, :, indices] > 33
# 替换标记位置的元素为22
arr[mask] = 22

方案二：直接修改子数组（更高效） #

利用Numpy切片的视图特性，直接提取第三轴指定索引的子数组，修改其中满足条件的元素：

import numpy as np

arr = np.random.rand(10, 200, 50) * 100
indices = [0, 1, 15, 20, 19]

# 提取目标子数组（视图，修改会同步到原数组）
sub_arr = arr[:, :, indices]
# 替换子数组中大于33的元素为22
sub_arr[sub_arr > 33] = 22

方案二更简洁高效，因为它只操作需要修改的部分，无需处理整个大数组的掩码。如果indices包含重复值，两种方案的最终结果一致，只是方案二会重复修改对应位置，但不影响最终效果。

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