机器学习(五) 关于散点图生成

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)