numpy中where函数的用法是什么

avatar
作者
筋斗云
阅读量:0

numpy中的where函数用于根据给定的条件从原数组中选择元素。它的用法如下:

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

参数:

  • condition:需要满足的条件,可以是一个布尔数组或布尔表达式。
  • x和y:可选参数,表示满足条件的元素替换为x中的元素,不满足条件的元素替换为y中的元素。x和y可以是一个数组或标量。

返回值:

  • 返回一个新的数组,形状与原数组相同。

示例:

import numpy as np  arr = np.array([1, 2, 3, 4, 5]) new_arr = np.where(arr > 3, arr, 0) print(new_arr) # 输出:[0 0 0 4 5] 

在上面的示例中,arr > 3 是一个布尔数组,满足条件的元素为True,不满足条件的元素为False。根据条件,将满足条件的元素替换为arr中的对应元素,不满足条件的元素替换为0,得到了新的数组new_arr。

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!