numpyのarray

Question

以下でA[A>=B]がarray([1, 2, 3, 4])といった１次元の配列になるのが理解できません。

Aは１×５
Bも１×５
なので
A[A>=B]も１×５になると思っていたのですが、１次元配列の(4,)でした。
これが理解できませんでした。
ご指導よろしくお願いいたします。

import numpy as np

A = np.array([[0, 1, 2, 3, 4]])
B = np.full((1, 5), 1)
A[A>=B]

Answer 1

それはnumpyの仕様です。
という答えでは納得がいかないでしょうね。

numpyを作っている人たちはこう考えたのでしょう。

ndarrayの要素の中で、ある条件を満たす要素だけを抜き出したいことがあるなあ。
そういう機能をつけようか。
でも、例えば(3,3)のshapeのndarrayで、1より大きい数を抜き出したら、条件を満たす要素の数は3個のときもあるし、7個のときもある。それを2次元の配列で返すのは無理だ。
使い方を考えたら1次元の配列を返すようにするしかないし、それで困らないよね。

というようなことで、そういう仕様になっているのでしょう。

Answer 2

import numpy as np

A = np.array([[0, 1, 2, 3, 4]])
B = np.full((1, 5), 1)
A[A>=B]

上記コードでA>=Bの結果はarray([[False, True, True, True, True]])となります。
ですのでA[A>=B]は配列AからA>=BでTrueの要素を抽出するのでarray([1, 2, 3, 4])が結果となります。