Pythonで差がKより大きいタプルを削除する方法

Pythonで差がKより大きいタプルを削除する方法

リストに含まれるタプルのうち、要素間の差が特定の値Kより大きいタプルを削除したい場合があります。このような処理は、組み込み関数 abs() を使うことで簡単に実現できます。abs() は絶対値を返すため、タプル内の要素の順序に関係なく、正確に差を計算できるのがポイントです。

サンプルコード

my_tuple = [(41, 18), (21,57), (39, 22), (23, 42), (22, 10)]
print("The tuple is :")
print(my_tuple)
K = 20
my_result = [element for element in my_tuple if abs(element[0] - element[1]) <= K]
print("The result is :")
print(my_result)

出力結果

The tuple is :
[(41, 18), (21, 57), (39, 22), (23, 42), (22, 10)]
The result is :
[(39, 22), (23, 42), (22, 10)]

コードの解説

• まず、複数のタプルを含むリストを定義し、コンソールに表示します。

• 比較の基準となる値 K を定義します(ここでは 20)。

• リスト内包表記を使ってリストを反復処理し、各タプルの第1要素と第2要素の差の絶対値が K 以下であるかどうかを判定します。

• 条件を満たすタプルだけが新しいリストとして変数 my_result に代入されます。

• 最後に、フィルタリングされた結果をコンソールに表示します。

上記の出力を見ると、元のリストから (41, 18) と (21, 57) の2つのタプルが除外されていることがわかります。これらは要素間の差がそれぞれ 23 と 36 であり、K の値である 20 より大きいためです。

filter() 関数を使った別の書き方

同じ処理は、filter() 関数と lambda 式を組み合わせても実現できます。

my_result = list(filter(lambda element: abs(element[0] - element[1]) <= K, my_tuple))

どちらの方法でも結果は同じですが、リスト内包表記の方がPythonらしい簡潔な書き方として広く使われています。用途や可読性の好みに応じて使い分けるとよいでしょう。