指定した合計値となる長さKのグループを取得するPythonプログラム

特定の合計値となる長さ「K」のグループを取得したい場合、空のリストと、itertoolsモジュールの「product」メソッド、組み込みの「sum」メソッド、「append」メソッドを組み合わせて使用できます。

サンプルコード

以下に実際の動作例を示します。

from itertools import product

my_list = [45, 32, 67, 11, 88, 90, 87, 33, 45, 32]
print("The list is : ")
print(my_list)

N = 77
print("The value of N is ")
print(N)
K = 2
print("The value of K is ")
print(K)

my_result = []
for sub in product(my_list, repeat = K):
   if sum(sub) == N:
      my_result.append(sub)

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

出力

The list is :
[45, 32, 67, 11, 88, 90, 87, 33, 45, 32]
The value of N is
77
The value of K is
2
The result is :
[(45, 32), (45, 32), (32, 45), (32, 45), (45, 32), (45, 32), (32, 45), (32, 45)]

解説

• まず、必要なパッケージ（itertools）を環境にインポートします。

• 対象となるリストを定義し、コンソールに表示します。

• 目標となる合計値Nと、グループの長さKを定義し、コンソールに表示します。

• 結果を格納するための空のリストを用意します。

• productメソッド（repeat=K）を使うことで、リストの要素から重複を許して選んだ長さKのタプルをすべて生成し、その合計がNと一致するかどうかをsumメソッドで確認します。

• 条件に一致したタプルは、appendメソッドによって結果リストへ追加されます。

• 最後に、得られた結果をコンソールに出力します。

補足：重複を除去したい場合

出力例を見ると、(45, 32) や (32, 45) のように同じ組み合わせが複数回現れています。これは、productが順序を区別する組み合わせ（順列）を生成するためです。もし重複を取り除きたい場合は、set() を使って次のように書き換えることができます。

my_result = list(set(sub for sub in product(my_list, repeat=K) if sum(sub) == N))
print(my_result)

このようにすることで、順序の異なる同一組み合わせが1つにまとめられ、より見やすい結果を得られます。なお、要素数が多いリストや大きなKに対しては、productが生成する組み合わせの総数が急激に増えるため、処理時間に注意が必要です。