что здесь делает операция X [y == 0,1] в коде?

Question

В этом коде:

from sklearn.datasets import make_moons
import matplotlib.pyplot as plt

X, y = make_moons(n_samples=100, random_state=123)
plt.scatter(X[y==0,0], X[y==0,1], color='red', marker='^', alpha=0.5)
plt.scatter(X[y==1,0], X[y==1,1], color='blue', marker='o', alpha=0.5)
plt.show()

Что именно операция X[y==0,0] делает с набором данных X?

Answer 1

В этом случае обе переменные X и y имеют тип numpy .ndarray.

И, глядя на переменные, мы видим, что переменная X имеет форму (100, 2). Это означает, что X имеет 100 строк и 2 столбца. Точно так же y имеет форму (100,) и, таким образом, является просто одномерным вектором. Бывает только 0 и 1.

Таким образом, X[y==0,0] находит все строки X, имеющие y значение 0 (y == 0) и находящиеся в первом столбце X