python - numpy插入一个元素，保留数组维数

Question

我有一个像这样的 numpy 数组:

[[1, 2, 3], [1, 2, 4]]

我想将一个元素 [100, 101, 102] 添加到数组，如下所示:

[[1, 2, 3], [1, 2, 4], [100, 101, 102]]

我尝试了numpy.append，但它创建了一个包含所有元素的一维数组。我怎样才能做到这一点？

Answer 1

使用np.append时需要指定axis，并且值需要具有正确的形状；作品如下:

a = [[1, 2, 3], [1, 2, 4]]
b = [100, 101, 102]

np.append(a, [b], axis=0)
#array([[  1,   2,   3],
#       [  1,   2,   4],
#       [100, 101, 102]])

如果您有列表:

a.append(b)
np.array(a)

应该更有效率。