Python：如何在数组中查找特定元素的索引？

Question

鉴于此，我有以下数组：

 import numpy as np

 dt = np.array([1,2,3,4,5,2,1,3])

我可以通过以下代码选择值小于3的单元格：

print(dt[dt<3])

但是，我如何获得所选单元格的索引？

我的最爱结果是：

[0,1,5,6]

Answer 1

尝试

x = np.array([1,2,3,4,5,2,1,3])
np.where(x<3)

输出：

(array([0, 1, 5, 6], dtype=int64),)

您将获得所有正确的索引。

Answer 2

我不确定您是否需要numpy。

lst = [1, 2, 3, 4, 5, 2, 1, 3]
indexes = [i for i, v in enumerate(lst) if v < 3]