将一个 numpy 数组附加到一个列表 - 奇怪的事情

Question

在 Raspberry Pi 上的 Raspbian 上使用 Spyder 3.1.3 中的 Python3.5.3。 将两个 numpy 数组附加到名为 'list0' 的列表中可以很好地与分配的 numpy 数组 'a' 类似：

import numpy as np

list0 = []
a = np.array([[1,2,3],[2,3,4]])
list0.append(a)
a = np.array([[11,12,13],[12,13,14]])
list0.append(a)

print("list0 =",list0)

效果很好，作为输出提供（帖子的格式更好）：

list0 = [ array([[ 1,  2,  3], [ 2,  3,  4]]), 
          array([[11, 12, 13], [12, 13, 14]]) ]

使用循环将赋值替换为a，奇怪的事情发生了：

import numpy as np
a = np.empty((3), int)
list0 = []
for idx in range(4):    
    for i in range(3):
        a[i] = idx*10 + i
    print("idx =",idx,"; a =",a)
    list0.append(a)
print("list0 =",list0)

第二行告诉 Python 使用的数组的形状（在我原来的例子中，它是一个三维数组）。为了验证生成的名为“a”的数组被打印出来。将新填充的数组 'a' 附加到 'list0' 最终显示最后一行的四倍。

idx = 0 ; a = [ 0  1  2]
idx = 1 ; a = [10 11 12]
idx = 2 ; a = [20 21 22]
idx = 3 ; a = [30 31 32]
list0 = [ array([30, 31, 32]), array([30, 31, 32]), 
          array([30, 31, 32]), array([30, 31, 32]) ] 

我认为“list0”只包含四倍的指向数组“a”的指针，该数组仅存在于一个实例/内存范围中。

那么：如何将每个不同的数组 'a' 物理附加（复制？）到列表中？它是一个 python 错误还是只是我对某些东西的误解？当然，我应该多想pythonian ;c)

谢谢你的帮助，彼得

Answer 1

问题

您将相同的数组 a 附加到您的 list0 4 次。像a 这样的数组是可变对象，这意味着当你给它们赋值时，底层对象会发生变化。由于该数组在您的列表中出现了 4 次，因此这些更改（似乎）出现在 4 个不同的地方。

解决方案

您只需稍作改动即可修复您拥有的代码。将数组的副本附加到列表中，而不是数组本身：

import numpy as np
a = np.empty((3), int)
list0 = []
for idx in range(4):    
    for i in range(3):
        a[i] = idx*10 + i
    print("idx =",idx,"; a =",a)
    list0.append(a.copy())
print("list0 =",list0)

输出：

idx = 0 ; a = [0 1 2]
idx = 1 ; a = [10 11 12]
idx = 2 ; a = [20 21 22]
idx = 3 ; a = [30 31 32]
list0 = [array([0, 1, 2]), array([10, 11, 12]), array([20, 21, 22]), array([30, 31, 32])]

优化方案

Python/Numpy 提供了许多更好的方法来初始化数组（无论是使用更少的代码行还是运行速度更快）。对于像这样的一堆范围，这是一个合理的方法：

list0 = [np.arange(n*10, n*10+3) for n in range(4)]
print(list0)

输出：

[array([0, 1, 2]), array([10, 11, 12]), array([20, 21, 22]), array([30, 31, 32])]

您也可以考虑只使用单个二维数组来代替数组列表。单个数组通常比列表中的异构数组更易于使用。这样做的方法如下：

arr0 = np.array([np.arange(n*10, n*10+3) for n in range(4)])
print(arr0)

输出：

[[ 0  1  2]
 [10 11 12]
 [20 21 22]
 [30 31 32]]

Answer 2

这样做：

list_to_append.append(np_array.copy())

简而言之，numpy 数组或列表是可变对象，这意味着当您将 numpy 数组或列表分配给变量时，您真正分配的是什么是对内存位置的引用，也就是指针。

在您的情况下，“a”是一个指针，因此您真正要做的是将地址附加到 list0 到“a”指向的内存位置，而不是指针指向的实际值。 因此，这意味着“list0”的每个新位置，在附加之后，结果都是相同的内存地址：“a”。

所以，而不是：

list0.append(a)

你调用 "a" 的 copy() 方法为 "a" 的新值创建一个新的内存位置并返回它：

list0.append(a.copy())