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] 的数组。