使用np.extract多条件提取数组元素报错，求单行解决方案

这个问题我之前踩过坑！你遇到的ValueError是因为在NumPy里不能直接用Python原生的and/or来连接数组级别的条件判断——and会试图把整个布尔数组转换成单个布尔值，而数组有多个元素时，这个转换的“真值”是模糊的，所以就抛出了这个错误。

给你几个单行实现的解决方案，都是元素级别的逻辑判断，一次性完成多条件筛选：

方案1：使用按位与运算符&（推荐，更简洁） #

注意要给每个条件单独加括号，避免运算符优先级的问题：

import numpy as np

x = np.array([1,2,3,4])
x_sel = x[(x <= 4) & (x > 1)]
# 或者用np.extract的写法：
# x_sel = np.extract((x <= 4) & (x > 1), x)

方案2：使用NumPy内置逻辑函数np.logical_and #

如果觉得运算符不够直观，用NumPy提供的逻辑函数也可以，它专门处理数组的元素级逻辑与：

x_sel = x[np.logical_and(x <= 4, x > 1)]
# 对应np.extract的写法：
# x_sel = np.extract(np.logical_and(x <= 4, x > 1), x)

两种方法运行后，x_sel都会得到array([2, 3, 4])，完全符合你的需求，不用分两次处理~

内容的提问来源于stack exchange，提问作者Jan SE