def get_leafs(self, use_labels=False):
""" Use to retrieve labels.
Parameters
----------
use_labels : bool, optional
If True, self.labels will be returned. If False, will
use either columns (if matrix is pandas DataFrame)
or indices (if matrix is np.ndarray)
"""
if isinstance(self.mat, pd.DataFrame):
if use_labels:
return [self.labels[i] for i in scipy.cluster.hierarchy.leaves_list(self.linkage)]
else:
return [self.mat.columns.tolist()[i] for i in scipy.cluster.hierarchy.leaves_list(self.linkage)]
else:
return scipy.cluster.hierarchy.leaves_list(self.linkage)
评论列表
文章目录