假设我有一个 numpy 数组,如下所示:
我该如何提取一个包含小于6的元素索引的数组,以获得以下结果:
import numpy as np
arr = np.array([10,1,2,5,6,2,3,8])
我该如何提取一个包含小于6的元素索引的数组,以获得以下结果:
np.array([1,2,3,5,6])
我希望有一个类似于np.nonzero()的函数,但不是测试非零值,而是测试小于x的值。
np.nonzero
。 - DivakarFalse
值会被解释为零。 - Paul H