python numpy 多条件筛选

numpy 条件筛选

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

import numpy as np

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

python numpy 多条件筛选_第1张图片

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

python numpy 多条件筛选_第2张图片
这里可以缩写为

a[a>5]

多条件筛选

但是如果我们有多个条件时,比如 7>a>5 筛选,按照我的想法应该就是
a[7>a>5]
但是这里运行会报错:
在这里插入图片描述
经过一通查阅发现这里并不能这样写

应该写成

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

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

你可能感兴趣的:(python小工具,python)