What does X[y==yi] (Line: 18) do ?
This is the full code:
This is the full code:
from sklearn.datasets import load_iris import matplotlib.pyplot as plt import numpy as np iris = load_iris() X = iris.data[:, [0, 2]] #print(iris.data) y = iris.target def plot_scatter(X,y): colors = ["red","blue","black","yellow","green","purple","orange"] markers = ('s', 'x', 'o', '^', 'v') for i, yi in enumerate(np.unique(y)): Xi = X[y==yi] print(Xi) plt.scatter(Xi[:,0], Xi[:,1], color=colors[i], marker=markers[i], label=yi) plt.xlabel('X label') plt.ylabel('Y label') plt.legend(loc='upper left') plot_scatter(X,y)