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

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

条件を満たす要素抽出

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

書式

np.where(条件式)

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

ソースコード

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

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

# 配列の宣言
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) # 

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