NumPyで条件に応じた処理を行うnp.whereの使い方 | note.nkmk.me
numpy.where()を使うと、NumPy配列ndarrayの条件を満たす要素・満たさない要素を置換したり、特定の処理を適用したりできる。条件を満たす要素のインデックス(位置)を取得することも可能。 numpy.where — NumPy v ...