numpy 数组的列表索引如何工作？

Question

我有一小段代码，我正在尝试了解如何使用另一个 numpy 数组作为另一个 numpy 数组的索引。

bmi = np.array([13,14,15,22])
light = np.array(bmi) < 21
print(bmi[light])

输出：[13,14,15]

我知道，当使用 numpy 数组时，事情是按元素完成的，所以我对 bmi[light] 的假设是它只会打印 bmi 中对应的值，这些值在光照下的值为 true。

Answer 1

如果你想打印light，你会得到：

[ True  True  True False]

这是一个布尔掩码。

现在，当您使用 print(bmi[light]) 时，您的 bmi 过滤为 True。

您可以使用 ~ 反转该掩码以仅获得 [22]：

print(bmi[~light])

与 cmets 中指出的相同，您可以了解有关布尔掩码 here 的更多信息。