阅读量: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。