numpy中where函数的用法是什么-开发者知识库平台

numpy中where函数的用法是什么

numpy

2046

2024/1/13 10:02:09

栏目: 编程语言

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

numpy.where(condition[, 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。

numpy中where函数的用法是什么