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)