← retour aux snippets

scipy.cluster: linkage et dendrogram

Clustering hiérarchique et coupe par distance.

objectif

Clustering hiérarchique et coupe par distance.

code minimal

import numpy as np
from scipy.cluster.hierarchy import linkage, fcluster

X = np.array([[0,0],[0,1],[5,5]])
Z = linkage(X, method="ward")
labels = fcluster(Z, t=2, criterion="maxclust")
print(set(labels) <= set([1,2]))

utilisation

import numpy as np
from scipy.cluster.hierarchy import linkage

X = np.array([[0,0],[0,1],[5,5]])
Z = linkage(X, method="single")
print(Z.shape[1])

variante(s) utile(s)

# from scipy.cluster.hierarchy import dendrogram
# dendrogram(Z)
print("ok")

notes

  • ward pour clusters sphériques; single chaîne facile.