python限制itertools结果

Question

我使用itertools来获取列表的所有组合：

import itertools

stuff = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
for L in range(0, len(stuff)+1):
  for subset in itertools.combinations(stuff, L):
    print(subset)

有没有办法将结果限制为至少包含2个元素，最多包含5个元素的组合？

谢谢!

Answer 1

只需更改for循环。

import itertools

stuff = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
for L in range(2, 6):
  for subset in itertools.combinations(stuff, L):
    print(subset)

Answer 2

将范围限制为[2, 6)

for L in range(2, 6):
   for subset in itertools.combinations(stuff, L):
      print(subset)