Python SciPy hierarchy.cut_tree用法及代碼示例

本文簡要介紹 python 語言中 scipy.cluster.hierarchy.cut_tree 的用法。

用法:

scipy.cluster.hierarchy.cut_tree(Z, n_clusters=None, height=None)#

給定一個鏈接矩陣 Z，返回切割樹。

參數 ：：

Z： scipy.cluster.linkage 數組

聯動矩陣。

n_clusters： 數組，可選

樹中切點處的簇數。

height： 數組，可選

砍樹的高度。僅適用於超度量樹。

返回 ：：

cutree： 數組

在每個聚集步驟中指示組成員身份的數組。即，對於完整的切割樹，在第一列中，每個數據點都在自己的集群中。下一步，合並兩個節點。最後，所有單例和非單例集群都在一組中。如果給出了 n_clusters 或 height，則列對應於 n_clusters 或 height 的列。

例子：

>>> from scipy import cluster
>>> import numpy as np
>>> from numpy.random import default_rng
>>> rng = default_rng()
>>> X = rng.random((50, 4))
>>> Z = cluster.hierarchy.ward(X)
>>> cutree = cluster.hierarchy.cut_tree(Z, n_clusters=[5, 10])
>>> cutree[:10]
array([[0, 0],
       [1, 1],
       [2, 2],
       [3, 3],
       [3, 4],
       [2, 2],
       [0, 0],
       [1, 5],
       [3, 6],
       [4, 7]])  # random