python的numpy使用小技巧

一、通过条件筛选numpy数组并获得对应的值

        要获得numpy数列中满足条件的某些元素，不用通过for循环来获取，可以直接通过numpy.where()来寻找，并且通过获得的索引反向获取元素！

import numpy as np

a = np.array([1,5,6,8,2,3,5,1,9,0])

idx = np.where(a>5)[0]  # 获得大于5的元素的索引

a_subset = a[idx]  # 获得原数组中大于5的元素