[Python]リストの要素を条件で検索するには？

どうも、ちょげ(@chogetarou)です。

リスト(List)の要素を条件で検索する方法を紹介します。

方法

リスト(List)の要素を条件で検索する方法は、２つあります。

filter()

ひとつは、filter()を使う方法です。

まず、filter()を呼び出します。

filter()の第１引数に１つの引数を持つ関数、第２引数に対象のリストを指定します。

そして、第１引数の関数で条件式を返します。

条件式では、関数の引数でリストの要素を取得します。

#ls=対象のリスト, funcition=条件式を返す関数（１つの引数を持ち、引数で要素を取得）
result = filter(function, ls)

上記のfilter()は、関数で返した条件式で検索し、条件式を満たしたリストの要素を全てfilterオブジェクトとして取得します。

もし、filter()の結果をリストとして取得したい場合は、filter()の結果をlist()の引数に指定します。

#結果をリストとして取得
result = list(filter(function, ls))

使用例

nums = [1, 2, 3, 4, 5, 6]

result = list(filter(lambda x: x % 2 == 0, nums))

print(result)

出力：
[2, 4, 6]

内包表記

もうひとつは、内包表記を使う方法です。

まず、内包表記で、リストをfor文でループし、ループ変数を結果として返します。

そして、for文の後ろに、if文で要素の条件式を指定します。

if文の条件式では、for文のループ変数でリストの要素を取得します。

#list=対象のリスト, condition=要素の条件式
#条件式ではfor文のループ変数（x）でリストの要素を取得
result = [x for x in list if condition]

上記の内包表記は、リストをif文の条件式で検索し、条件式を満たすリストの要素を全て取得します。

使用例

nums = [1, 2, 3, 4, 5, 6]

result = [x for x in nums if x % 2 == 0]

print(result)

出力：
[2, 4, 6]

まとめ

リスト(List)の要素を条件で検索する方法は、次の２つです。

• filter()を使う方法
  result = filter(function, ls)
• 内包表記を使う方法
  result = [x for x in list if condition]