温馨提示×

numpy中where函数的用法是什么

小亿
94
2024-01-25 10:02:09
栏目: 编程语言

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。

0