Pythonでリストのすべてのサブリストを生成して出力する方法

リストが与えられたとき、そのリストに含まれるすべてのサブリスト(部分リスト)を生成して出力するプログラムを紹介します。要素数 n のリストの場合、空のリストを含めるとサブリストの総数は n×(n+1)/2+1 個になります。

実行例

入力 : list = [1, 2, 3]
出力 : [], [1], [1, 2], [1, 2, 3], [2], [2, 3], [3]

アルゴリズム

考え方はシンプルです。開始位置 i と終了位置 j の組み合わせをすべて試し、スライスで部分リストを取り出します。

ステップ1 : 対象となるリストを用意します。
ステップ2 : 結果を格納するため、最初に空のリストを含むリストを用意します。
ステップ3 : リストの長さ分だけ外側の for ループを回します。
ステップ4 : 内側のループを i+1 からリストの長さまで実行し、i より右側のすべての範囲を取得します。
ステップ5 : スライス A[i:j] で部分リストを切り出します。
ステップ6 : 切り出した部分リストを結果用のリストに追加します。
ステップ7 : 最後に結果をまとめて出力します。

サンプルコード

# Python program to print all
# sublist from a given list
# function to generate all the sub lists
def displaysublist(A):
   # store all the sublists
   B = [[]]

   # first loop
   for i in range(len(A) + 1):
      # second loop
      for j in range(i + 1, len(A) + 1):
         # slice the subarray
         sub = A[i:j]
         B.append(sub)
   return B

# driver code
A = list()
n = int(input("Enter the size of the First List ::"))
print("Enter the Element of First List ::")
for i in range(int(n)):
   k = int(input(""))
   A.append(k)
print("SUBLIST IS ::>", displaysublist(A))

コードのポイント

• range(len(A) + 1) により、開始位置 i を 0 からリストの長さまで動かしています。
• 内側の range(i + 1, len(A) + 1) で終了位置 j を決めることで、A[i:j] のスライスだけで連続した部分リストを簡単に取得できます。
• 初期値として空のリスト [] を入れておくことで、空のサブリストも結果に含まれます。

出力結果

Enter the size of the First List :: 3
Enter the Element of First List ::
1
2
3
SUBLIST IS ::> [[], [1], [1, 2], [1, 2, 3], [2], [2, 3], [3]]

このように、二重ループとスライスを組み合わせるだけで、リストのすべての連続するサブリストを効率よく生成できます。順列や組み合わせのように並び替えを伴うものではなく、「元のリストの連続した区切り」をすべて列挙する処理である点に注意してください。