大数据基础--鸢尾花

鸢尾花

import numpy as np
import matplotlib.pyplot as plt
import matplotlib as mpl
import os
from sklearn import datasets
path=os.getcwd()
iris=datasets.load_iris()
print(iris.get('feature_names'))
print(iris.get('target').shape)
X=iris.data[:,:2]
print(X.shape)
plt.scatter(X[:,0],X[:,1])
path1=path+'\\'+'test1.png'
plt.savefig(path1)
plt.show()
y=iris.target
plt.scatter(X[y==0,0],X[y==0,1],color='red')
plt.scatter(X[y==1,0],X[y==1,1],color='blue')
plt.scatter(X[y==2,0],X[y==2,1],color='green')
path2=path+'\\'+'test2.png'
plt.savefig(path2)
plt.show()