假设我有一个 numpy 数组,例如:
import numpy as np
arr = np.array([10,1,2,5,6,2,3,8])
Run Code Online (Sandbox Code Playgroud)
如何提取包含小于 6 的元素索引的数组,以便得到以下结果:
np.array([1,2,3,5,6])
Run Code Online (Sandbox Code Playgroud)
我想要一些行为类似于np.nonzero() 的东西,但不是测试非零值,而是测试小于 x 的值