python - Python如何查找数组中元素的位置？

Question

我有以下数组:

scores=[2.619,3.3, 9.67, 0.1, 6.7,3.2]

我希望通过以下代码检索超过 5 个的元素:

min_score_thresh=5
Result=scores[scores>min_score_thresh]

因此这将导致我的结果:

[9.67, 6.7]

现在我希望得到这两个元素的位置，这是我预期的答案将存储在变量 x 中:

x = [2,4]

请分享我的想法，谢谢

Answer 1

将 numpy.where 用于矢量化解决方案:

import numpy as np

scores = np.array([2.619,3.3, 9.67, 0.1, 6.7,3.2])
min_score_thresh = 5

res = np.where(scores>min_score_thresh)[0]

print(res)

[2 4]