sklearn.PCA(2)를 이용한 digit data visualization

pca=PCA(2)
X_pca=pca.fit_transform(mnist.data)

plt.figure(figsize=[10,8])
plt.scatter(X_pca[:,0],X_pca[:,1],c=mnist.target)
plt.colorbar()

 

xmax,ymax=X_pca.max(axis=0)
xmin,ymin=X_pca.min(axis=0)

colors = ["#476A2A", "#7851B8", "#BD3430", "#4A2D4E", "#875525",
          "#A83683", "#4E655E", "#853541", "#3A3120", "#535D8E"]
plt.figure(figsize=[14,14])
plt.xlim([xmin,xmax])
plt.ylim([xmin,xmax])
for i in range(len(X_pca)):
    plt.text(X_pca[i,0],X_pca[i,1],str(mnist.target[i]),color=colors[mnist.target[i]],fontdict={'weight':'bold','size':9})