numpy的就地操作（例如，“ + =”）如何工作？

Question

基本问题是：执行以下操作时会发生什么a[i] += b？

给定以下内容：

import html" target="_blank">numpy as np
a = np.arange(4)
i = a > 0
i
= array([False,  True,  True,  True], dtype=bool)

我明白那个：

• a[i] = x与相同a.__setitem__(i, x)，直接分配给所指示的项目i
• a += x与相同a.__iadd__(x)，即会进行加法

但是当我这样做时会发生什么 ：

a[i] += x

特别：

1. 这和一样a[i] = a[i] + x吗？（这不是就地操作）
2. 如果i是，在这种情况下是否有所不同：
   • 一个int指数，或
   • 一个ndarray或
   • 一个slice对象

背景

我开始研究此问题的原因是在使用重复索引时遇到了非直觉的行为：

a = np.zeros(4)
x = np.arange(4)
indices = np.zeros(4,dtype=np.int)  # duplicate indices
a[indices] += x
a
= array([ 3.,  0.,  0.,  0.])

Answer 1

您需要意识到的第一件事是，a += x它并不完全映射到a.__iadd__(x)，而是映射到a = a.__iadd__(x)。请注意，文档特别指出，就地运算符返回其结果，而不必这样做self（尽管实际上通常是这样）。这意味着a[i] += x简单地映射到：

a.__setitem__(i, a.__getitem__(i).__iadd__(x))

因此，添加从 技术上讲 是就地进行的，但仅在临时对象上进行。但是__add__，创建的临时对象可能比调用的对象少一个。