python3 既存コードの拡張

Question

現状
3枚のカードを並べて、1からxの組み合わせをl配列にいれました。
合計がYのカードに組み合わせを出力してます。

拡張したい部分
3枚ではなく、T枚のカードを並べるには、
どの様なコードが適切なのでしょうか？

失敗コード

l=[]
r=[]
def Sample02(x,y):
	for i in range(1,x):
		for i2 in range(1,x):
			for i3 in range(1,x):
				l.append([i,i2,i3])
	for i in l:
		if sum(i)==y:
			r.append(i)

Sample02(20,20)
print(r)

zなどを加えて、for _ in range()
の回数を増やす事は可能でしょうか？

考え方が違って居たなら指摘下さい。

Comments

[退会済みユーザー]

1からxまでの要素を取り出すにはrange(1, x+1)としないと、右端のxが含まれなくなります。

Answer 1

こんな感じでしょうか。

from itertools import combinations

def sample(x, y, t=3):
    return [
        nums for nums in combinations(range(1, x), t) if sum(nums) == y
    ]

print(*sample(20, 20), sep='\n')

実行結果 Wandbox

---

コメントでご指摘を受けている通り、重複ありの組み合わせの方が適しているかもしれません。
実行結果(重複なし／重複あり)を見てご判断ください。

Comments

[LouiS0616]

なお、元のコードと同一な処理をしたい場合、『組み合わせ』ではなく『順列』となります。
その場合mkgreiさんやdkato0077さんのご回答が参考になるかと。

[退会済みユーザー]

重複を許さない組み合わせならcombinations, 重複を許すならcombinations_with_replacementですね。

[LouiS0616]

あ、確かにcombinations_with_replacementの方がよさそうですね。
コメントありがとうございます。

[SpecialTukimiya]

なるほど。

Answer 2

itertoolsモジュールのproductを使うと比較的簡単に書けます。

from itertools import product
x = 10
y = 5
T = 3
lst = range(1, x+1)
r = []
for l in product(lst, repeat=T):
    if sum(l) == y:
        r.append(l)

なお、このfor-loopはリスト内包表記に置きかえることができます。

r = [l for l in product(lst, repeat=T) if sum(l) == y]

Answer 3

itertoolsのproductはいかがでしょう。
https://docs.python.jp/3/library/itertools.html#itertools.product