np.where的奇怪用法

时间:2019-09-24 13:53:18

标签: python numpy

我正在查看别人的代码,这些代码大量使用了numpy。在一种情况下,代码显示如下内容:

>>> a = np.array([True, True, False, False, True])
>>> np.where(a)
(array([0, 1, 4], dtype=int64),)

我已经使用过np.where,但没有以这种方式使用。我检查了该函数的文档,但没有找到可以将布尔区域直接传递给该函数的地方,但这仍然可行。

具有更多麻木经验的人可以解释一下这里到底发生了什么吗?该输出数组代表什么?谢谢

2 个答案:

答案 0 :(得分:2)

您只是得到数组值是True的索引

答案 1 :(得分:0)

在发布此问题的2秒内,我想我才发现它。

我认为它只是提供数组值为True的索引。