如何在python中仅使用“O”或“Z”打印给定长度的所有序列？

Question

如何在Python中仅使用'O'或'Z'打印给定长度的所有序列？你知道吗

例如，如果我给length = 3

输出为：

'OOO'
'OOZ'
'OZO'
'OZZ'
'ZOO'
'ZZO'
'ZOZ'
'ZZZ'

Answer 1

可以通过使用itertools生成元素的笛卡尔积来实现这一点：

import itertools

letters = ["O", "Z"]
length = 3

vals = [''.join(comb) for comb in itertools.product(letters, repeat=length)]

print(vals)
>>>['OOO', 'OOZ', 'OZO', 'OZZ', 'ZOO', 'ZOZ', 'ZZO', 'ZZZ']