NumPy教程-numpy.where()在 Python 中的使用

语法:

numpy.where(condition[, x, y])

参数:

专属福利 
👉点击领取：最全Python资料合集

返回值:

示例 1: np.where()

import numpy as np  
a=np.arange(12)  
b=np.where(a<6,a,5*a)  
b  

在上面的代码中

• 我们导入了别名为 np 的 numpy 库。

• 我们使用 np.arange() 函数创建了一个名为 'a' 的数组。

• 我们声明了变量 'b'，并将 np.where() 函数的返回值赋给它。

• 我们在函数中传递了数组 'a'。

• 最后，我们尝试打印 'b' 的值。

输出:

array([ 0,  1,  2,  3,  4,  5, 30, 35, 40, 45, 50, 55])

示例 2: 对于多维数组

import numpy as np  a=np.arange(12)  b=np.where([[True, False], [True, True]],[[1, 2], [3, 4]],[[9, 8], [7, 6]])  b

输出:

array([[1, 8],        [3, 4]])

示例 3: 广播 x、y 和 condition

import numpy as np  x, y = np.ogrid[:3, :4]  a=np.where(x > y, x, 10 + y)  a  

输出:

array([[10, 11, 12, 13],        [ 1, 11, 12, 13],        [ 2,  2, 12, 13]])

在上面的代码中

• 我们导入了别名为 np 的 numpy 库。

• 我们使用 np.arange() 函数创建了一个名为 'a' 的数组。

• 我们声明了变量 'b'，并将 np.where() 函数的返回值赋给它。

• 我们在函数中传递了一个布尔值的多维数组作为条件，并传递了整数数组 x 和 y。

• 最后，我们尝试打印 'b' 的值。

示例 4: 广播特定值

x=np.array([[0,1,2],[0,2,5],[0,4,8]])  y=np.where(x<4,x,-2)  y

输出:

array([[ 0,  1,  2],       [ 0,  2, -2],       [ 0, -2, -2]])

 1

热门推荐

• NumPy教程-numpy.log()在 Python 中的使用

• 这5种炫酷的动态图，都是用Python实现的！

• NumPy教程-numpy.zeros() 在 Python 中的使用