我有一个像这样的 numpy 数组:
[[1, 2, 3], [1, 2, 4]]
我想将一个元素 [100, 101, 102]
添加到数组,如下所示:
[[1, 2, 3], [1, 2, 4], [100, 101, 102]]
我尝试了numpy.append
,但它创建了一个包含所有元素的一维数组。我怎样才能做到这一点?
最佳答案
使用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)
应该更有效率。
关于python - numpy插入一个元素,保留数组维数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/44857278/