Numpy 'where' 为条件 >= 0 的所有 0 数组返回空数组

Question

我正在使用 np.where 函数来获取值与 0 匹配的索引。但以下代码返回空数组，这是不期望的。

import numpy as np
a = np.array([0,0,0,0])
np.where(a[a>=0])

这给出： (array([], dtype=int64),) 。

谁能指出我缺少什么？

Answer 1

这是预期的。 a>=0给你所有True ，即a[a>=0]给你a ，它不包含任何非零元素。 所以np.where(a)返回空数组。

你在找np.where(a>=0)吗？

Answer 2

你告诉找到索引，其中 a[ a>=0] a>=0 返回 [True,True,True,True] 然后 a[a>=0] 返回 [0,0,0,0] 然后没有返回的真实条件。

也许你想做

 np.where(a>=0)