使用 numpy argmax 在以下数组中查找大于 5 的值的第一次出现： arr = range(2, 20)

Question

Use numpy argmax to find the first occurrence of value greater than 5 in the following array: arr = range(2, 20)

Answer 1

import numpy as np

arr = np.array(range(2, 20))
idx = np.argmax(np.concatenate((arr[arr <= 5], [arr[arr > 5][0]])))

or

import numpy as np

arr = np.array(range(2, 20))
idx = np.argmax(arr[arr <= 5]) + 1