使用 itertools 生成 2 个 1 和 3 个 0 的所有排列

Question

可能是基本的，但在任何其他问题中都找不到。 我试过了：

print ["".join(seq) for seq in itertools.permutations("00011")]

但是有很多重复，似乎 itertools 不理解所有的零并且所有的都是相同的......

我错过了什么？

编辑：

哎呀。感谢 Gareth，我发现这个问题是重复的：permutations with unique values。 没有关闭它，因为我认为我对问题的措辞更清楚。

Answer 1

list(itertools.combinations(range(5), 2))

返回一个包含 10 个位置的列表，其中两个位置可以在五位数以内（其他位置为零）：

[(0, 1),
 (0, 2),
 (0, 3),
 (0, 4),
 (1, 2),
 (1, 3),
 (1, 4),
 (2, 3),
 (2, 4),
 (3, 4)]

对于有 2 个 1 和 13 个 0 的情况，请使用：

list(itertools.combinations(range(5), 2))

返回一个包含 105 个位置的列表。而且它比您原来的解决方案要快得多。

现在的功能：

def combiner(zeros=3, ones=2):
    for indices in itertools.combinations(range(zeros+ones), ones):
        item = ['0'] * (zeros+ones)
        for index in indices:
            item[index] = '1'
        yield ''.join(item)

print list(combiner(3, 2))

['11000',
 '01100',
 '01010',
 '01001',
 '00101',
 '00110',
 '10001',
 '10010',
 '00011',
 '10100']

这需要 14.4µs。

list(combiner(13, 2))

返回 105 个元素需要 134µs。

Answer 2

set("".join(seq) for seq in itertools.permutations("00011"))