有没有办法用 Python 计算一个集合的所有合法分区？

Question

Wiki 给出了这个集合的分区的definition

在数学中，集合的分区是将集合的元素分组为非空子集，这样每个元素都包含在一个子集中。

还有这个例子

集合{1,2,3}有这五个分区

{ {1}, {2}, {3} }, sometimes written 1|2|3.
{ {1, 2}, {3} }, or 12|3.
{ {1, 3}, {2} }, or 13|2.
{ {1}, {2, 3} }, or 1|23.
{ {1, 2, 3} }, or 123

有没有办法用 Python 计算一个集合的所有合法分区？

我试过the partitions in sympy

from sympy.combinatorics.partitions import Partition
a = Partition([1, 2, 3])
a.members

我得到了

(1, 2, 3)

这显然是不正确的。

Answer 1

如果您使用 Sympy，那么您需要 sympy.utilities.iterables.multiset_partitions:

>>> from sympy.utilities.iterables import multiset_partitions
>>> for p in multiset_partitions([1, 2, 3]):
...     print(p)
...
[[1, 2, 3]]
[[1, 2], [3]]
[[1, 3], [2]]
[[1], [2, 3]]
[[1], [2], [3]]