Python SciPy hierarchy.DisjointSet用法及代码示例

本文简要介绍 python 语言中 scipy.cluster.hierarchy.DisjointSet 的用法。

用法:

class  scipy.cluster.hierarchy.DisjointSet(elements=None)#

用于增量连接查询的不相交集数据结构。

注意：

这个类实现了不相交集[1]，也称为union-find或者merge-find数据结构。这找操作(实施于__getitem__) 实现路径减半变体。这合并方法实现按大小合并变体。

参考：

[1]

https://en.wikipedia.org/wiki/Disjoint-set_data_structure

例子：

>>> from scipy.cluster.hierarchy import DisjointSet

初始化一个不相交集：

>>> disjoint_set = DisjointSet([1, 2, 3, 'a', 'b'])

合并一些子集：

>>> disjoint_set.merge(1, 2)
True
>>> disjoint_set.merge(3, 'a')
True
>>> disjoint_set.merge('a', 'b')
True
>>> disjoint_set.merge('b', 'b')
False

查找根元素：

>>> disjoint_set[2]
1
>>> disjoint_set['b']
3

测试连通性：

>>> disjoint_set.connected(1, 2)
True
>>> disjoint_set.connected(1, 'b')
False

列出不相交集中的元素：

>>> list(disjoint_set)
[1, 2, 3, 'a', 'b']

获取包含 ‘a’ 的子集：

>>> disjoint_set.subset('a')
{'a', 3, 'b'}

获取包含‘a’的子集的大小(无需实际实例化子集)：

>>> disjoint_set.subset_size('a')
3

获取不相交集中的所有子集：

>>> disjoint_set.subsets()
[{1, 2}, {'a', 3, 'b'}]

属性 ：：

n_subsets： int

子集的数量。