一、通过条件筛选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的元素
要获得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的元素