y==0是bool运算,X[y==0, 0] 表示把y==0为真时X第一列的数据全取出来
则取出来的全是0这一类对应的样本数据。
通过下面这个例子应该就明白了
import numpy as npimport matplotlib.pyplot as pltfrom sklearn import datasetsiris = datasets.load_iris()print(iris.keys())X = iris.data[:,:2]# plt.scatter(X[:,0], X[:,1])# plt.show()y = iris.targetplt.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()