如何在不使用if语句的情况下对选定索引的Numpy数组执行加法？

没问题，这事儿用Numpy的向量化索引就能轻松搞定，完全不需要写if语句～

为什么你的代码不行？ #

你当前的np.add(x, y[idx])会触发Numpy的广播错误：因为x是长度为5的一维数组，而y[idx]是长度为2的一维数组，两者的维度无法匹配广播规则，运行时会抛出类似ValueError: operands could not be broadcast together with shapes (5,) (2,)的错误，自然得不到你想要的结果。

正确的实现方式 #

核心思路是：先复制原数组x作为结果的基础，再通过索引定位到需要修改的位置，直接加上y中对应位置的元素。具体代码如下：

import numpy as np

x = np.array([1., 2., 3., 4., 5.])
y = np.array([1., 1., 1., 1., 1.])
idx = np.array([1, 3])

# 先复制x，避免修改原数组
res = x.copy()
# 只对idx指定的位置执行相加操作
res[idx] += y[idx]

print(res)

运行这段代码会输出：

[1. 3. 3. 5. 5.]

补充说明 #

这种方法完全依赖Numpy的向量化操作，比循环遍历加if判断的效率高得多，尤其是处理大规模数组时优势明显。如果y的元素不是全1，这个方法依然通用——它会精准地把x中idx位置的元素和y中idx位置的元素相加，其他位置保持不变。

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