numpy.subtract带where参数时输出异常问题咨询

核心是numpy ufunc的where参数的设计行为 #

np.subtract属于numpy的通用函数（ufunc），它的where参数的规则是：

• 仅在where条件为True的位置执行减法计算，将结果写入输出数组对应位置
• where条件为False的位置，输出数组对应位置不会被修改，保留初始化时的原始值

当你没有显式指定out参数时，numpy会自动创建一个和输入数组形状、数据类型完全一致的新数组作为输出容器，但这个新数组是未初始化状态，它的初始值就是内存中对应地址的残留数据，没有任何默认值（不是0，也不是输入数组的值）。

两种场景的差异解释 #

仅调用带where的np.subtract的场景 #

此时申请的内存块刚好是未被使用过的，残留值全为0，所以where为False的位置就保留了初始的0，和你预期的结果一致，但这是内存分配的巧合，不属于numpy承诺的稳定行为。

先调用不带where的np.subtract的场景 #

1. 不带where的np.subtract执行时，会申请一块内存存储输出结果，值为你打印的[[ 0 -1],[ 0 0], ...]
2. 这个输出数组打印完成后没有其他引用，会被numpy的内存池回收
3. 后续调用带where的np.subtract时，numpy申请新输出数组的内存，刚好复用了刚才被回收的内存块，所以新数组的初始值就是之前不带where的计算结果
4. where为False的位置不会被覆盖，就保留了之前的-1等旧值，最终输出就和不带where的结果一致。

稳定复现预期结果的写法 #

如果需要where为False的位置保留原始数组的值，必须显式初始化输出数组：

import numpy as np

exponents = np.indices((4, 4))[:, ::2].reshape((2, -1)).T
# 先把输出数组初始化为原始值
out = exponents.copy()
# 仅在符合条件的位置执行减法，直接写入out数组
np.subtract(out, [0, 1], where=exponents > 0, out=out)
print(out)

或者直接用np.where函数写逻辑更清晰：

out = np.where(exponents > 0, exponents - [0, 1], exponents)

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