Python numpy数组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数组：与基本Python相比，它是NumPy非常糟糕的一个操作。 这是因为您正在制作每个append数据的完整副本，这将花费您二次时间。

相反，只需将您的数组附加到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)