np.where
是NumPy库中一个非常实用的函数,它允许我们根据指定的条件在数组中进行元素选择和替换。在本篇文章中,我们将详细解析np.where
的用法,并提供一些示例代码来说明其实际应用。
1. np.where
函数概述
np.where
函数的语法如下:
np.where(condition, x, y)
其中,condition
是一个布尔型数组或条件表达式,表示我们要应用的条件。x
和y
是两个数组或者标量,它们的形状可以不一样。当condition
中的元素为真时,np.where
将返回x
中对应位置的元素;当condition
中的元素为假时,将返回y
中对应位置的元素。
np.where
函数的返回值是一个新的数组,其形状与输入的<