Kmean(简单例子)

l1 = [88,74,96,85]

l2 = [92,99,95,94]

l3 = [91,87,99,95]

l4 = [78,99,97,81]

l5 = [88,78,98,84]

l6 = [100,95,100,92]

X = np.array([l1,l2,l3,l4,l5,l6])
X = np.mat(X)

sklearn

from sklearn.cluster import KMeans
import numpy as np

kmeans = KMeans(n_clusters=2).fit(X)
kmeans.predict(X)
# array([1, 0, 0, 0, 1, 0], dtype=int32)

kmeans.predict([[ 192,  100,  85,  94]])
# array([0], dtype=int32)

Scipy 聚类

from scipy.cluster.vq import vq,kmeans,whiten

whiten = whiten(X)
cen,c = kmeans(whiten,2)
vq(whiten,cen)
# (array([0, 1, 1, 0, 0, 1], dtype=int32),
# array([1.28751842, 1.8732278 , 1.05869903, 1.90840468, 0.97381232,
#        1.49659288]))