Pythonで条件に基づいてリスト内のすべての組み合わせを抽出する方法

ネストされたリスト構造から、指定した条件に従ってすべての組み合わせを取り出したい場面はよくあります。Pythonでは、シンプルな反復処理（whileループとforループ）、appendメソッド、そしてisinstance関数を組み合わせることで、この処理を簡単に実現できます。

実装例

以下のコードは、文字列とリストが混在するリストから、各サブリストの同じインデックス位置にある要素を組み合わせて、新しいリストを生成する例です。

my_list = ["python", [15, 12, 33, 14], "is", ["fun", "easy", "better", "cool"]]

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

K = 4
print("The value of K is :")
print(K)

my_result = []
count = 0
while count <= K - 1:
   temp = []

   for index in my_list:

      if not isinstance(index, list):
         temp.append(index)
      else:
         temp.append(index[count])
   count += 1
   my_result.append(temp)

print("The result is :")
print(my_result)

出力結果

The list is :
['python', [15, 12, 33, 14], 'is', ['fun', 'easy', 'better', 'cool']]
The value of K is :
4
The result is :
[['python', 15, 'is', 'fun'], ['python', 12, 'is', 'easy'], ['python', 33, 'is', 'better'], ['python', 14, 'is',
'cool']]

コードの解説

• まず、文字列とリストが混在したリストmy_listを定義し、コンソールに表示します。

• 生成する組み合わせの数を決める値Kを定義し、コンソールに表示します。

• 結果を格納するための空のリストmy_resultを作成します。

• カウンタとして使用する変数countを作成し、0で初期化します。

• whileループでcountがK未満である間、リスト内の各要素を順番に走査します。このとき、isinstance関数を使って、要素がリスト型かどうかを判定します。

• 要素がリストでない場合はそのまま一時リストtempに追加し、リストの場合は現在のcount番目の要素を取り出して追加します。これにより、各サブリストから同じ位置の要素を組み合わせたリストが完成します。

• 1回分の組み合わせができあがったら、それをmy_resultに追加し、countを1増やして次の組み合わせの生成へ進みます。

• 最終的に、すべての組み合わせが格納された結果がコンソールに表示されます。

ポイント：isinstance関数とは

isinstance(オブジェクト, 型)は、第一引数のオブジェクトが第二引数で指定した型（またはそのサブクラス）のインスタンスであるかどうかを判定するPythonの組み込み関数です。本記事の例では「要素がリストかどうか」を判別するために活用されており、異なる型が混在するデータ構造を柔軟に処理する際に非常に役立ちます。