numpy扩展模块,用于有效搜索比较为true的第一个数组索引
py-find-1st的Python项目详细描述
功能性
这个扩展解决了查找第一个索引的常见问题 无需读取完整阵列。
通话顺序
import numpy as np import utils_find_1st as utf1st limit = 0. rr= np.random.randn(100) ind = utf1st.find_1st(rr < limit, True, utf1st.cmp_equal)
而且效率更高
ind = utf1st.find_1st(rr, limit, utf1st.cmp_smaller)
相当于
import numpy as np limit = 0. rr= np.random.randn(100) ind = np.flatnonzero(rr < limit) if len(ind) : ret = ind[0] else: ret = -1