关于Numpy高级索引中x[[0,0]] +=1执行逻辑的疑问

核心差异：两种写法的结果不同 #

先看实际运行结果：

import numpy as np
x = np.arange(2)
x[[0,0]] += 1
print(x)  # 输出 [2 0]

x = np.arange(2)
x[[0,0]] = x[[0,0]] + 1
print(x)  # 输出 [1 0]

这两种写法的结果差异，就是问题的根源——+=并没有按常规的“取值-计算-赋值”流程执行。

为什么调用顺序是getitem→iadd→setitem #

Python中obj[a] += b的语法糖，本质遵循以下标准流程：

1. 调用v = obj.__getitem__(a)获取目标对象
2. 调用v = v.__iadd__(b)执行原地加法
3. 调用obj.__setitem__(a, v)将结果写回

这就是你的子类测试代码打印getitem→iadd→setitem的原因，这是Python复合赋值的通用执行顺序，并非NumPy特有的隐式绑定。

NumPy的特殊优化：直接累积修改原数组 #

关键在于NumPy对__iadd__和高级索引的特殊处理：

• 当通过高级索引（比如整数列表[0,0]）获取数组时，__getitem__返回的确实是原数组的拷贝，但NumPy在__iadd__方法中做了优化——它不会真的修改这个拷贝，而是直接遍历索引列表，对原数组的每个对应位置执行累积加法。
• 换句话说，x[[0,0]] +=1等价于执行两次x[0] +=1，而非先创建[0,0]的拷贝、改成[1,1]再写回原数组。

而普通赋值x[[0,0]] = x[[0,0]] +1则是严格按“取值-计算-赋值”流程：先拿到[0,0]的拷贝，计算得到[1,1]，再把这个数组写回原索引位置——也就是两次将1赋值给x[0]，最终结果自然是1。

关于setitem的作用 #

你的测试代码中打印了setitem，但这个步骤在+=的场景下几乎是一个“收尾操作”，因为实际的修改已经在__iadd__阶段完成了。NumPy保留这个调用只是为了符合Python的语法糖规范，并非真的需要将拷贝的值写回原数组。

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