【NumPy】配列から条件を満たす要素を抽出

スポンサーリンク

この記事では、Python言語とNumPyを用いて、配列(リスト)から特定の条件を満たす要素を抽出する方法をソースコード付きで解説します。

スポンサーリンク

条件を満たす要素抽出

NumPyの「numpy.where」を利用することで、指定した条件を満たす配列(リスト)の要素番号を取り出すことが出来ます。

書式

np.where(条件式)

今回は、配列y内の数値データのうち、10以上20未満のデータがある要素を取り出す。

ソースコード

サンプルプログラムのソースコードです。

# -*- coding: utf-8 -*-
import numpy as np

def main():

    # 配列の宣言
    y = np.array([8,9,10,11,15,18,22,21,20,29]) 

    # 条件式(10以上20未満)を満たす要素を抽出
    y2 = y[ np.where( (y>=10) & (y<20)) ]

    # 結果を表示
    print(y2)


if __name__ == '__main__':
    main()

実行結果

サンプルプログラムの実行結果です。

[10 11 15 18]

【おすすめ記事】
Python入門 サンプル集
NumPy入門 サンプル集

スポンサーリンク

シェア&フォローお願いします!