Python3入门机器学习经典算法与应用 第3章 读取数据和简单的数据探索

y==0是bool运算，X[y==0, 0] 表示把y==0为真时X第一列的数据全取出来

则取出来的全是0这一类对应的样本数据。

通过下面这个例子应该就明白了

 

 

 

 

import numpy as np
import matplotlib.pyplot as plt
from sklearn import datasets



iris = datasets.load_iris()
print(iris.keys())

X = iris.data[:,:2]
# plt.scatter(X[:,0], X[:,1])
# 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")
plt.show()