如何将行添加到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)
是否有 numpythonic的 方法可以做到这一点?
谢谢,S ;-)
什么X啊 如果它是一个二维数组,你怎么能那么其行比作一个号码:i < 3?
X
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
< 3
import numpy as np A = np.vstack((A, X[X[:,0] < 3])) # returns: array([[0, 1, 2], [0, 2, 0], [0, 1, 2], [1, 2, 0], [2, 1, 2]])