'List [条件语句]'此代码应该做什么

Question

（在for循环之后）请解释语句 X [y == yi] 。

代码：

    :

    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