在 Python 中设置分区

Question

我有一个数组 [1,2,3]

我想使用数组的所有元素进行所有可能的组合：

结果：

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

原文由 user2880257 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

由于这个很好的问题已经复活，这里有一个新的答案。

问题递归解决：如果你已经有了一个 n-1个 元素的分区，你如何用它来分区 n个 元素？要么将第 n 个元素放入现有子集中，要么将其添加为新的单独子集。仅此而已；没有 itertools ，没有集合，没有重复输出，总共只有 n 次调用 partition() ：

 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

something = list(range(1,5))

for n, p in enumerate(partition(something), 1):
    print(n, sorted(p))

输出：

 1 [[1, 2, 3, 4]]
2 [[1], [2, 3, 4]]
3 [[1, 2], [3, 4]]
4 [[1, 3, 4], [2]]
5 [[1], [2], [3, 4]]
6 [[1, 2, 3], [4]]
7 [[1, 4], [2, 3]]
8 [[1], [2, 3], [4]]
9 [[1, 3], [2, 4]]
10 [[1, 2, 4], [3]]
11 [[1], [2, 4], [3]]
12 [[1, 2], [3], [4]]
13 [[1, 3], [2], [4]]
14 [[1, 4], [2], [3]]
15 [[1], [2], [3], [4]]

原文由 alexis 发布，翻译遵循 CC BY-SA 4.0 许可协议