基于条件替换多维NumPy数组中的元素时赋值失效问题咨询

嘿，这个问题我之前也碰到过！核心原因是你用的是numpy的高级索引，这种方式返回的是原数组的副本，而不是视图——所以你后续对这个子数组的赋值操作，只是修改了临时生成的拷贝，原数组a根本没被改动。

具体来说：

• 当你用布尔数组（比如a[:, 0] == 1）或者整数数组去索引numpy数组时，numpy会创建原数据的拷贝，而不是像切片（比如a[0:2, :]）那样返回原数组的视图（视图是原数组的引用，修改视图会直接影响原数组）。
• 你写的a[a[:, 0] == 1][:, 2] = [9999, 9999]，其实是先通过a[a[:,0]==1]创建了一个副本，然后修改这个副本的第三列，最后这个副本用完就被丢弃了，原数组完全没碰着。

正确的解决方案 #

直接用组合索引定位到要修改的元素，这样就能在原数组上原地修改：

# 直接定位第一列为1的行的第三列，赋值
a[a[:, 0] == 1, 2] = [9999, 9999]

或者更清晰一点，先定义掩码再使用：

mask = a[:, 0] == 1
a[mask, 2] = 9999  # 这里也可以直接写9999，numpy会自动广播到符合条件的所有位置

执行完上面的代码后，你再打印a就能得到预期的结果：

print(a)
# 输出：
# [[   1    2 9999]
#  [   1   20 9999]
#  [   4    5    6]]

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