PythonでK桁の要素を持つタプルを抽出する方法

特定の桁数を持つ要素を含むタプルだけを抽出したい場合、リスト内包表記を使うと簡潔に実装できます。リスト内包表記は、タプルのリストを順番に反復処理しながら、指定した条件を評価します。これにより、条件を満たす要素のみがフィルタリングされ、新しい変数に格納されます。

サンプルコード

my_list = [(34, 56), (45, 6), (111, 90), (11, 35), (78, )]

print("The list is : ")
print(my_list)

K = 2
print("The value of K has been initialized to " + str(K))

my_result = [sub for sub in my_list if all(len(str(elem)) == K for elem in sub)]

print("The tuples extracted are : ")
print(my_result)

出力結果

The list is :
[(34, 56), (45, 6), (111, 90), (11, 35), (78,)]
The value of K has been initialized to 2
The tuples extracted are :
[(34, 56), (11, 35), (78,)]

処理の流れと解説

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

• 変数「K」に目標となる桁数（ここでは2）を設定します。

• リスト内包表記を使い、タプルのリストを1つずつ反復処理します。

• 各要素を str() で文字列に変換し、その長さ（＝桁数）がKと一致するかどうかを all() 関数で判定します。タプル内のすべての要素がK桁である場合のみ、そのタプルが抽出対象になります。

• 条件を満たしたタプルだけを集めた新しいリストを生成し、変数に代入します。

• 最終的な抽出結果をコンソールに出力して完了です。

この例では、(45, 6) は「6」が1桁のため除外され、(111, 90) は「111」が3桁のため除外されます。一方、(34, 56)・(11, 35)・(78,) はすべての要素が2桁なので、結果として抽出されています。なお、(78,) のように要素が1つでも、その要素がK桁であれば問題なく抽出される点に注意してください。