代码之家 › 专栏 › 技术社区 › itthrill

在中使用numpy.选择条件

numpy python

itthrill · 技术社区 · 6 年前

x = np.arange(10)
condlist = [x in [2,3,4], x>5]
choicelist = [x, x**2]
np.select(condlist, choicelist)

有办法让它工作吗?

1 回复 | 直到 6 年前

PMende 6 年前

你应该使用 isin 取而代之的是:

x = np.arange(10)
condlist = [np.isin(x, [2,3,4]), x>5]
choicelist = [x, x**2]
np.select(condlist, choicelist)

array([ 0,  0,  2,  3,  4,  0, 36, 49, 64, 81])

推荐文章

serlingpa · 如何准备我的数据以避免无法推断频率

1 年前

Guillaume · 使用操作从Python列表创建numpy数组

2 年前

user19657580 · 在Python中打印两个numpy数组的列表

2 年前

user19657580 · Python中数组中具有相同元素的索引求和

2 年前

mikanim · 改进二维余弦函数的numpy功能

2 年前

Klimt865 · 在Python中将数组列表转换为列表列表

2 年前

theduker · 计算平均绝对误差时,If语句中赋值前引用的局部变量

2 年前

Lynn · 如果列包含Python中的特定字符串,则从列中删除值

2 年前

JasonX · 运行减法计算

2 年前

Jan Hrubec · 选择numpy数组的前n个元素

2 年前