' List[ Conditional statement ] ' What is this code supposed to do.

[Arindam]

What does X[y==yi] (Line: 18) do ?

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)