给定一个数字,生成长度为n的X和O的所有可能组合的列表

Question

例如,如果function(2)那时它会产生

['XX','XO','OX','OO']

我不确定如何解决这个问题

Answer 1

除非这是一个赋值,否则请使用productitertools,如下所示:

>>> import itertools as it
>>> list(it.product('XO', repeat=2))
[('X', 'X'), ('X', 'O'), ('O', 'X'), ('O', 'O')]

如果您想知道如何以"长"方式执行此操作product,则文档中提供了该方法的实现:

def product(*args, **kwds):
    # product('ABCD', 'xy') --> Ax Ay Bx By Cx Cy Dx Dy
    # product(range(2), repeat=3) --> 000 001 010 011 100 101 110 111
    pools = map(tuple, args) * kwds.get('repeat', 1)
    result = [[]]
    for pool in pools:
        result = [x+[y] for x in result for y in pool]
    for prod in result:
        yield tuple(prod)