从数组中提取符合条件的元素
arr = np.arange(10)
arr
#array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
#设置条件,如:能整除3
condition = np.mod(arr, 3)==0
condition
#array([ True, False, False, True, False, False, True, False, False,True])
#np.extract()会把True对应的元素提取出来返回
np.extract(condition, arr)
#输出:array([0, 3, 6, 9])
参考:
[1] Extract:从数组中提取符合条件的元素
746



被折叠的 条评论
为什么被折叠?



