python 基础笔记(小技巧-利用布尔值抽取数据)

问题导入：

有数组如下，要求吧大于5的数据找出来

array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

解决思路：

首先来个条件判断提取，发现返回值是bool类型

 再变身一下下：(只可意会，妙哉！)

附上代码测试：

import numpy as np
a=np.arange(10)
print(a>5)
a[a>5]