numpy找到数组中符合条件的数
import numpy as np
arr = np.array([1, 1, 1, 134, 45, 3, 46, 45, 65, 3, 23424, 234, 12, 12, 3, 546, 1, 2])
print(arr[arr > 3])
Output:
[ 134 45 46 45 65 23424 234 12 12 546]
参考 https://blog.youkuaiyun.com/blackyuanc/article/details/77948703