Numpy数组第一行指定元素修改出错，求原因及解决方案

你的代码出错是因为np.where(p[0]==0)返回的是第一行中0所在的列索引（这里是array([2])），直接用p[np.where(...)]会选中整个第三行（数组索引从0开始），所以才会把第三行全部改成100。

要只修改第一行的目标位置，需要明确指定行索引+列索引的组合：

import numpy as np

p = np.array([[2,3,0,5],[2,3,4,5],[2,3,4,5],[0,0,0,0]])
# 直接定位第一行的目标列索引并赋值
p[0, np.where(p[0]==0)[0]] = 100

print(p)

运行后就能得到预期输出：

[[  2   3 100   5]
 [  2   3   4   5]
 [  2   3   4   5]
 [  0   0   0   0]]

除了np.where，还有几种更直观的方式：

1. 直接布尔索引定位 #

不用np.where，直接用布尔数组筛选第一行的0值位置：

p[0, p[0] == 0] = 100

或者更简洁的写法（效果完全一致）：

p[0][p[0] == 0] = 100

2. 使用np.putmask #

np.putmask可以直接对指定区域的掩码位置赋值，逻辑更清晰：

np.putmask(p[0], p[0] == 0, 100)

3. 遍历修改（仅作参考，不推荐） #

如果是新手想理解底层逻辑，可以用循环，但这种方式效率远低于numpy的向量操作，不适合处理大数组：

for i in range(len(p[0])):
    if p[0][i] == 0:
        p[0][i] = 100

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