NumPy入門！whereで条件に応じた要素の抽出

条件を指定しての配列からの要素の抽出は、zipを利用してもできますが、NumPyのwhereを利用すると簡単に記述することができます。この記事では、zip、whereの使い方とその違いを確認していきましょう。

動画教材紹介私(清水義孝)が作成したコース「Pythonによるビジネスに役立つWebスクレイピング」(Udemyへのリンク)が発売中！
GWに新スキルを習得しよう！セールは４月２１日まで！GW前の大セール中！

NumPyでのzipによる条件指定での抽出

同じ要素数のndarray型の配列A、B、conditionを用意します。配列A、Bについては適当な数値を格納し、配列conditionにはTrue、もしくは、Falseを格納します。配列A、B、conditionの要素を順に読み込み、conditionの値がTrueの場合はAから値を取得し、conditionの値がFalseの場合はBから値を取得するプログラムを記述します。

In [1]: A=np.array([1,2,3,4])
...: B=np.array([100,200,300,400])
...: condition=np.array([True,True,False,False])
...: answer=[(A_val if cond else B_val) for A_val,B_val,cond in zip(A,B,condition)] ...: answer

Out[1]: [1, 2, 300, 400]

zipでは複数の配列などのオブジェクトの要素を取り出すことができ、'for A_val,B_val,cond in zip(A,B,condition)'では、'zip(A,B,condition)'で配列A、B、conditionの要素を順に取り出し、それぞれ変数A_val、B_val、condに格納していきます。これらの変数は'A_val if cond else B_val'のif文でcondがTrueであれば変数A_valの値、condがFalseであれば変数B_valの値を出力用の配列の要素としています。