python - 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 和 2 处具有 0 元素。因此，X[y == 0] 为您提供一个包含 X[0] 和 X[2] 的数组。