使用numpy argmax在以下数组中查找第一个大于5的值：arr=range(2，20)
uxhixvfz1#
import numpy as np arr = np.array(range(2, 20)) idx = np.argmax(np.concatenate((arr[arr <= 5], [arr[arr > 5][0]])))
或
import numpy as np arr = np.array(range(2, 20)) idx = np.argmax(arr[arr <= 5]) + 1
1条答案
按热度按时间uxhixvfz1#
或