NumPy二维数组条件索引赋值问题：实际结果与预期不符

你的代码没有正确选中所有符合条件行的指定列，导致结果不符合预期。

问题原因 #

原代码中arr[(arr[:,0] < 4), [0,2]]的索引逻辑有问题：布尔数组[True, True]（来自arr[:,0] <4）和列索引数组[0,2]组合时，NumPy会将它们广播为匹配的索引对，最终只选中了(0,0)和(1,2)这两个单独元素，而非所有符合条件行的第0、2列，因此赋值结果不符合需求。

解决方案 #

方法一：用np.ix_构造正确的行/列索引

np.ix_可以将行、列索引数组转换为可广播的索引网格，确保选中所有符合条件行的指定列：

import numpy as np
arr = np.array([[0,1,2],[3,4,5]])
mask = arr[:, 0] < 4
arr[np.ix_(mask, [0, 2])] = 10
print(arr)

运行结果：

[[10  1  10]
 [10  4  10]]

方法二：分开对指定列赋值（直观易懂）

直接针对符合条件的行，分别给第0列和第2列赋值：

import numpy as np
arr = np.array([[0,1,2],[3,4,5]])
mask = arr[:, 0] < 4
arr[mask, 0] = 10
arr[mask, 2] = 10
print(arr)

运行结果与预期完全一致。

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