python - 在循环中切片 NumPy 数组

Question

<分区>

Answer 1

第一个例子:

您提取一行并将其加 1。然后您重新定义 指针 行 但不是数组 包含的内容!所以不会影响原来的数组。

第二个例子:

您进行就地操作 - 显然这会影响原始数组 - 只要它是一个数组。

如果你在做一个双循环，它就不会再工作了:

def example4(array):
    for row in array:
        for column in row:
            column += 1
    return array

example4(np.arange(9).reshape(3,3))
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])

这不起作用，因为您不调用 np.ndarray 的 __iadd__(修改数组指向的数据)，而是调用 python int 的 __iadd__。所以这个例子只适用于你的行是 numpy 数组。

第三个例子:

row[:] = row + 1 这被解释为类似于 row[0] = row[0]+1, row[1] = row[1]+1 , ... 同样，这会在原地工作，因此会影响原始数组。

底线

如果您正在对可变对象进行操作，例如 list 或 np.ndarray，您需要小心更改。这样的对象仅指向实际数据在内存中的存储位置 - 因此更改此指针(example1)不会影响保存的数据.您需要跟随指针(直接通过 [:] (example3) 或间接通过 array.__iadd__ (example2)) 更改保存的数据。