我如何使用itertools解决此问题？

Question

您好，我尝试使用此工具：

list(combinations('01', 3))

但是我得到这个结果：

[]

我希望获得这些结果：

000
001
011
111
101
100
010

另外，我也想有不同的情况。

例如，给定111，我期望得到这些结果：

111
12
21
3

是否可以使用itertools做这两件事？

Answer 1

这里是一个解决方案，部分使用了this answer中的解决方案。

• 来自Alexis答案的分区函数（请参阅链接）返回划分成组的列表（即[[[1,1,1]]，[[1,1]，[1]]，[[1]，[1,1]]，[[1]，[1 ]，[1]]]
• 为了得到您的问题中的数字，我添加了仅存储每个内部列表之和的部分
• list（dict.fromkeys（...））用于删除重复项

代码

def partition(collection):
    if len(collection) == 1:
        yield [ collection ]
        return

    first = collection[0]
    for smaller in partition(collection[1:]):
        # insert `first` in each of the subpartition's subsets
        for n, subset in enumerate(smaller):
            yield smaller[:n] + [[ first ] + subset]  + smaller[n+1:]
        # put `first` in its own subset 
        yield [ [ first ] ] + smaller


elements = list([1,1,1])

lst = list(dict.fromkeys([ tuple([sum(p[1][i]) for i in range(len(p[1]))]) for p in enumerate(partition(elements), 1) ]))

print(sorted(lst))

输出：

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

Answer 2

您似乎在问两个独立的问题。根据您的结果：

1. 00000101101111110101010

通过permutations_with_replacement使用itertools.product组合。

2. 111 12 21 3

通过itertools.product使用partitions算法。

给出

more_itertools.partitions

代码

more_itertools.partitions

import itertools as it

import more_itertools as mit

---

详细信息

为了获得特定的结果，请使用列表理解：

# 1 - Permutation w Replacement
>>> list(it.product("01", repeat=3))
[('0', '0', '0'),
 ('0', '0', '1'),
 ('0', '1', '0'),
 ('0', '1', '1'),
 ('1', '0', '0'),
 ('1', '0', '1'),
 ('1', '1', '0'),
 ('1', '1', '1')]

# 2 - Partitions
>>> list(mit.partitions("111"))
[[['1', '1', '1']],
 [['1'], ['1', '1']],
 [['1', '1'], ['1']],
 [['1'], ['1'], ['1']]]

# 1
>>> ["".join(x) for x in it.product("01", repeat=3)]

['000', '001', '010', '011', '100', '101', '110', '111']

是第三方程序包。用

# 2
>>> [[sum(map(int, x)) for x in sub] for sub in mit.partitions("111")]

[[3], [1, 2], [2, 1], [1, 1, 1]]

安装。