scipy.cluster.hierarchy.cut_tree¶
- scipy.cluster.hierarchy.cut_tree(Z, n_clusters=None, height=None)[源代码]¶
在给定链接矩阵Z的情况下，返回切割后的树。
- 参数
- Zscipy.cluster.link数组
链接矩阵。
- n_clustersARRAY_LIKE，可选
剪切点处树中的簇数。
- heightARRAY_LIKE，可选
砍树的高度。只有超尺度树才有可能。
- 退货
- 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