1. ホーム
  2. パイソン

[解決済み】Numpy - 配列に行を追加する

2022-04-11 20:55:48

質問

numpyの配列に行を追加するにはどうすればよいですか?

配列Aを持っています。

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

Xの各行の最初の要素が特定の条件を満たした場合に、別の配列Xからこの配列に行を追加したい。

Numpyの配列にはリストのような 'append' というメソッドがないようです。

AとXがリストだったら、単にやるだけです。

for i in X:
    if i[0] < 3:
        A.append(i)

はありますか? ノンフィクション を行う方法はありますか?

ありがとうございます。 S ;-)

解決方法は?

とは X ? 2次元配列の場合、その行を数値と比較することができます。 i < 3 ?

OPのコメント後にEDIT。

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 :

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]])