python numpy 多条件筛选

numpy 条件筛选

np.where – array中满足条件的索引

import numpy as np

a = np.arange(10)
print(np.where(a>5))

print(a[np.where(a>5)])

这里可以缩写为

a[a>5]

多条件筛选

但是如果我们有多个条件时，比如 7>a>5 筛选，按照我的想法应该就是
a[7>a>5]
但是这里运行会报错：

经过一通查阅发现这里并不能这样写

应该写成

a[(a>5)&(a<9)]

注意，里面的括号必须带着
那如果是并集的话写法就是把 & 换为 |。