'List [Условное выражение]' Что должен делать этот код

Question

Пожалуйста, объясните утверждение X [y == yi] (после цикла for).

Код:

    :

    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) 

Answer 1

условный оператор оценивается как True или False.

list[True] работает так же, как list[1], а list[False] работает так же, как list[0]

>>> l = [0, 1]
>>> l[False]
0
>>> l[True]
1