import numpy as np #随机生成点 from sklearn.datasets import make_blobs #K-means:k均值聚类 cluster(一簇,一类) from sklearn.cluster import KMeans import matplotlib.pyplot as plt %matplotlib inline X_train,y_train = make_blobs(n_samples=150,centers=3,cluster_std=1) plt.scatter(X_train[:,0],X_train[:,1],c = y_train) #立体图生成 plt.figure(figsize=(9,9)) axes3d = plt.subplot(projection = '3d') axes3d.scatter3D(ball['2006世界杯'],ball['2010世界杯'],ball['2007亚洲杯'],c = y_,cmap = 'rainbow') cluster_centers_ = kmeans.cluster_centers_ axes3d.scatter3D(cluster_centers_[:,0],cluster_centers_[:,1],cluster_centers_[:,2], c = [-1,3,5],cmap = plt.cm.cool,s = 300,alpha = 0.5)