Python numpy数组

Question

我在创建numpy数组的numpy数组时遇到了一个问题。我会在一个循环中创建它：

a=np.array([])
while(...):
   ...
   b= //a numpy array generated
   a=np.append(a,b)
   ...

期望的结果：

[[1,5,3], [9,10,1], ..., [4,8,6]]

实际结果：

[1,5,3,9,10,1,... 4,8,6]

有可能吗？我不知道数组的最终维数，所以我不能用固定的维数初始化它。

Answer 1

永远不要在循环中附加到numpy数组:这是NumPy与基本Python相比非常不擅长的一个操作。这是因为您将在每个append中制作数据的完整副本，这将耗费您2倍的时间。

相反，只需将数组附加到Python列表并在末尾转换它；结果会更简单、更快：

a = []

while ...:
    b = ... # NumPy array
    a.append(b)
a = np.asarray(a)

至于为什么你的代码不能工作：np.append的行为完全不像list.append。特别是，它不会在追加时创建新的维度。您必须创建具有二维的初始数组，然后附加一个显式的轴参数。

Answer 2

我们也可以尝试一下：

arr1 = np.arange(4)
arr2 = np.arange(5,7)
arr3 = np.arange(7,12)

array_of_arrays = np.array([arr1, arr2, arr3])
array_of_arrays
np.concatenate(array_of_arrays)