查找列表元素的所有可能排列，直到组合达到指定的长度为止

Question

假设一个列表L=[a,b]，我想找到该列表的所有可能组合，并在组合达到特定长度时停止。例如：我想在组合达到长度2后停止它。结果应为['aa', 'ab', 'ba', 'bb']

Answer 1

• itertools.product(list, repeat=2)-返回输入可迭代项的笛卡尔积。

例如。

import itertools
a = ['a','b']
combination = [''.join(x) for x in itertools.product(a,repeat=2)]
print(combination)

O / P：

['aa', 'ab', 'ba', 'bb']