Numpy Проблема индексации ..... Предварительная индексация, что здесь делает X [0]?

Question

import numpy as np

X = np.array([[0, 1, 0, 1], [1, 0, 1, 1], [0, 0, 0, 1], [1, 0, 1, 0]])

y = np.array([0, 1, 0, 1])

counts = {}

print(X[y == 0])

# prints = [[0 1 0 1]
# [0 0 0 1]]

Я хочу знать, почему X[y==0] печатает две точки данных. Разве он не должен печатать только [0 1 0 1]?

, потому что X[0]?

Answer 1

y == 0 дает массив с такими же размерами, как y, с элементами True, где соответствующий элемент в y равен 0, а False в противном случае.

Здесь, y имеет 0 элементов с индексами 0 и 2. Итак, X[y == 0] дает вам массив, содержащий X[0] и X[2].