python - 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]

标签： pythonnumpynumpy-ndarray

Answer 1

尝试

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

输出：

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

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