python中where()的用法
import numpy as np
a= np.array([8,2,7,5,1,4])
print(a<5)
print(a[a<5])
print(np.where(a<5))
print(a[np.where(a<5)])
定义一个数组a，print(a)的输出结果是“布尔值索引”：
print(a[a<5])可以根据索引取出数据：
上面的方法可以用where()函数代替：print(np.where(a<5))，输出的是索引：
根据索引取数据：print(a[np.where(a<5)])
