Python中的颜色，标签

Question

X 是一个 100x2 向量，包含 2 种花的萼片和花瓣长度数据。 y 是一个 100x1 向量，其中包含我的数据的标签值：-1 和 1。我创建了一个网格网格，然后我使用 countourf 方法绘制了我的网格网格，现在我使用以下代码将数据加载到网格网格中：

for idx, cl in enumerate (np.unique(y)):
    plt.scatter (x=X[y == cl, 0], y= X[y == cl, 1], alpha=0.8, c=colors[idx], marker= markers [idx], label = cl, edgecolor = 'black')

alpha, colors, marker, edgecolor 只是次要的东西。还有np.unique(y) = [-1 1]。

我的问题是，为什么[y==cl,0] 是错误的，而[y==,1] 是正确的论点？还有，如何使用== 对我的数据进行分类？

Answer 1

在 for 循环中，cl 循环遍历y 的两个唯一 值，即[-1, 1]。因此，在第一次循环迭代中，cl=-1 和 y == cl 返回 y 值为 -1 的行。因此，[y == cl, 0] 返回 y 为 -1 的位置，数据是第一列（索引 0），即 Sepal 长度。同样，[y == cl, 1] 返回 y 为 -1 的位置，数据为第二列（索引 1），即花瓣长度。

cl 为 1 时，循环的第二次迭代也是如此。