将行添加到空的2D NumPy数组

时间:2014-11-21 13:38:35

标签: python numpy

我想从一个空的2D NumPy数组开始,然后向它添加一些行。但是,到目前为止,我只能用一维数组做到这一点。这是我到目前为止所尝试的:

a = numpy.array([])
a = numpy.append(a, [1, 2])
a = numpy.append(a, [8, 8])
print a

我得到的输出是:

[1, 2, 8, 8]

我希望输出为:

[[1, 2], [8, 8]]

我该怎么做?

2 个答案:

答案 0 :(得分:5)

试试这个:

>>> a = numpy.empty((0,2),int)
>>> a = numpy.append(a, [[1, 2]], axis=0)
>>> a = numpy.append(a, [[8, 8]], axis=0)
>>> a
array([[ 1,  2],
       [ 8,  8]])

答案 1 :(得分:1)

>>> import numpy
>>> numpy.vstack(([1, 2], [8, 8]))
array([[1, 2],
       [8, 8]])