np.where多种情况判定
时间: 2023-11-16 09:59:44 浏览: 88
python numpy之np.random的随机数函数使用介绍
5星 · 资源好评率100%
np.where是一个用于多种情况判定的函数,它可以根据给定的条件返回两个数组中相应位置的元素。np.where的语法如下:
np.where(condition, x, y)
其中,condition是一个布尔型数组,x和y是两个数组,它们的形状必须相同。当condition中的元素为True时,np.where返回x中相应位置的元素;当condition中的元素为False时,np.where返回y中相应位置的元素。
np.where的应用场景很多,例如可以用它来实现数据清洗、数据转换、数据筛选等操作。下面是一个例子:
假设有一个数组a,我们想将其中大于0的元素替换为1,小于0的元素替换为-1,等于0的元素不变,可以使用np.where实现:
a = np.array([1, 0, -1, 2, -3, 0])
b = np.where(a > 0, 1, np.where(a < 0, -1, 0))
print(b)
输出结果为:[ 1 0 -1 1 -1 0]
在上面的例子中,我们首先使用np.where(a > 0, 1, np.where(a < 0, -1, 0))判断a中每个元素的正负情况,然后根据情况返回1、-1或0,最终得到了一个新的数组b。
阅读全文