Numpy-如何将行添加到数组中?

内容来源于 Stack Overflow,并遵循CC BY-SA 3.0许可协议进行翻译与使用

  • 回答 (2)
  • 关注 (0)
  • 查看 (8)

如何将行添加到numpy数组?

我有一个阵列A:

A = array([[0, 1, 2], [0, 2, 0]])

如果X中每一行的第一个元素满足特定条件,我希望从另一个数组X向该数组添加行。

Numpy数组没有类似于列表的方法“追加”,至少看起来是这样的。

如果A和X是清单,我只会:

for i in X:
    if i[0] < 3:
        A.append(i)
提问于
用户回答回答于

X是吗?如果它是2D数组,那么如何将其行与数字进行比较:i < 3?

在OP的评论之后编辑:

A = array([[0, 1, 2], [0, 2, 0]])
X = array([[0, 1, 2], [1, 2, 0], [2, 1, 2], [3, 2, 0]])

加到A所有行X第一个元素< 3:

A = vstack((A, X[X[:,0] < 3]))

# returns: 
array([[0, 1, 2],
       [0, 2, 0],
       [0, 1, 2],
       [1, 2, 0],
       [2, 1, 2]])
用户回答回答于

你可以这样做:

  newrow = [1,2,3]
  A = numpy.vstack([A, newrow])

扫码关注云+社区