Pythonでタプルのリストから特定の要素を含むタプルを検索する方法

Pythonでは、リストの要素としてタプルを持つことができます。本記事では、タプルのリストの中から、特定の文字列要素を含むタプルを抽出する方法を2つのアプローチで解説します。

リスト内包表記と条件式を使う方法

リスト内包表記にif文による条件を組み合わせると、目的のタプルだけを簡潔に抽出できます。条件は単一でも、and や or を使った複数条件の組み合わせでも問題ありません。

サンプルコード

listA = [('Mon', 3), ('Tue', 1), ('Mon', 2), ('Wed', 3)]
test_elem = 'Mon'
# 元のリスト
print("Given list:\n",listA)
print("Check value:\n",test_elem)
# forとifを使用
res = [item for item in listA
    if item[0] == test_elem and item[1] >= 2]
# 結果を出力
print("The tuples satisfying the conditions:\n ",res)

実行結果

上記のコードを実行すると、以下の出力が得られます。

Given list:
[('Mon', 3), ('Tue', 1), ('Mon', 2), ('Wed', 3)]
Check value:
Mon
The tuples satisfying the conditions:
[('Mon', 3), ('Mon', 2)]

この例では、タプルの最初の要素が 'Mon' と一致し、かつ2番目の要素が2以上という両方の条件を満たすタプルのみが新しいリストに格納されます。

filter関数とlambdaを使う方法

filter() 関数とラムダ式（lambda）を組み合わせる方法もあります。filterの条件部分では、in 演算子を使ってタプル内に指定した要素が存在するかどうかを判定します。

サンプルコード

listA = [('Mon', 3), ('Tue', 1), ('Mon', 2), ('Wed', 3)]
test_elem = 'Mon'
# 元のリスト
print("Given list:\n",listA)
print("Check value:\n",test_elem)
# lambdaとinを使用
res = list(filter(lambda x:test_elem in x, listA))
# 結果を出力
print("The tuples satisfying the conditions:\n ",res)

実行結果

上記のコードを実行すると、以下の出力が得られます。

Given list:
[('Mon', 3), ('Tue', 1), ('Mon', 2), ('Wed', 3)]
Check value:
Mon
The tuples satisfying the conditions:
[('Mon', 3), ('Mon', 2)]

この方法では、タプルのどの位置に検索対象の要素があってもマッチするため、「要素が含まれているかどうか」だけを確認したい場合に便利です。位置を指定して厳密に判定したい場合は、リスト内包表記の方が柔軟に対応できます。