numpy where函数用法

numpy where

函数用法

Numpy where

函数是

Numpy

中的一个高级函数，在一些数学、统计和通

用计算任务中，它经常被用来替代三元表达式（

if-else

）。

Numpy where

函数根据条件检查要求，从输入列表（数组）中选择或过

滤不同的值，并返回相应的输出列表，其语法如下：

numpy.where(condition, x, y) 

其中，

condition

是将对输入数组执行的布尔函数；

x

和

y

是可选项，

表示将使用以下三个值中的一个：

（

1

）当

condition

条件为真时，返回

x

；（

2

）当

condition

条件为假

时，返回

y

；（

3

）如果条件与

x

和

y

均无关，则返回输入数组中的值。

使用

Numpy where

函数可以有效替换传统

if-else

语句进行数据选择，

有益于高效处理大量数据，同时也减少编程语言的复杂性。

例如，我们利用

where

函数来处理一个简单的问题：给定一个包含整

数的数组，要求根据给定的阈值，将数组值中小于阈值的值替换成

0

，

而大于阈值的值替换成

1. 

我们先来看看如何使用

if-else

函数进行处理：

    # 

为了演示，定义一个简单的数组

    arr= np.array([10, 20, 30, 40])