如何将新行添加到空的numpy数组中

Question

使用标准Python数组,我可以执行以下操作:

arr = []
arr.append([1,2,3])
arr.append([4,5,6])
# arr is now [[1,2,3],[4,5,6]]

但是,我不能在numpy做同样的事情.例如:

arr = np.array([])
arr = np.append(arr, np.array([1,2,3]))
arr = np.append(arr, np.array([4,5,6]))
# arr is now [1,2,3,4,5,6]

我也研究过vstack,但是当我vstack在一个空阵列上使用时,我得到:

ValueError: all the input array dimensions except for the concatenation axis must match exactly

那么如何在numpy中向空数组添加一个新行呢？

Answer 1

"启动"所需数组的方法是:

arr = np.empty((0,3), int)

这是一个空数组,但它具有适当的维度.

>>> arr
array([], shape=(0, 3), dtype=int64)

然后一定要沿轴0追加:

arr = np.append(arr, np.array([[1,2,3]]), axis=0)
arr = np.append(arr, np.array([[4,5,6]]), axis=0)

但是,@ jonrsharpe是对的.实际上,如果你要在循环中追加,那么在第一个例子中追加到列表会快得多,然后在最后转换为numpy数组,因为你真的没有使用numpy作为在循环期间的意图:

In [210]: %%timeit
   .....: l = []
   .....: for i in xrange(1000):
   .....:     l.append([3*i+1,3*i+2,3*i+3])
   .....: l = np.asarray(l)
   .....: 
1000 loops, best of 3: 1.18 ms per loop

In [211]: %%timeit
   .....: a = np.empty((0,3), int)
   .....: for i in xrange(1000):
   .....:     a = np.append(a, 3*i+np.array([[1,2,3]]), 0)
   .....: 
100 loops, best of 3: 18.5 ms per loop

In [214]: np.allclose(a, l)
Out[214]: True

numpythonic的方法取决于你的应用程序,但它更像是:

In [220]: timeit n = np.arange(1,3001).reshape(1000,3)
100000 loops, best of 3: 5.93 µs per loop

In [221]: np.allclose(a, n)
Out[221]: True

Answer 2

在这种情况下,您可能希望使用函数np.hstack和np.vstack

arr = np.array([])
arr = np.hstack((arr, np.array([1,2,3])))
# arr is now [1,2,3]

arr = np.vstack((arr, np.array([4,5,6])))
# arr is now [[1,2,3],[4,5,6]]

您还可以使用np.concatenate函数.

干杯

Answer 3

这是我的解决方案:

arr = []
arr.append([1,2,3])
arr.append([4,5,6])
np_arr = np.array(arr)