Python中np.where()的使用

最新推荐文章于 2025-05-30 17:30:29 发布

允诺@晴天

最新推荐文章于 2025-05-30 17:30:29 发布

阅读量5.9k

点赞数 4

CC 4.0 BY-SA版权

分类专栏： Python 文章标签： python numpy 开发语言

本文链接：https://blog.youkuaiyun.com/qq_52006948/article/details/131570966

Python 专栏收录该内容

5 篇文章

订阅专栏

np.where()是NumPy中的一个函数，用于在数组中根据条件选择元素。它接受一个布尔数组作为条件，以及两个替代值，返回的结果数组中，满足条件的元素被第一个值替换，否则被第二个值替换。文中给出了多个示例，包括替换满足条件的元素、获取满足条件的元素位置以及处理多个条件的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

np.where的使用

np.where()是NumPy库中一个非常有用的函数，用于根据指定的条件返回一个向量或数组中满足条件的元素的位置。它的基本语法是：

np.where(condition, x, y)

其中：

condition是一个布尔数组或布尔条件表达式，用于指定需要满足的条件。
x和y分别是满足条件和不满足条件时的替代值。它们可以是标量、向量或数组。

np.where()函数返回一个与condition大小相同的数组，其中满足条件的元素用x替代，不满足条件的元素用y替代。
下面是几个示例：

import numpy as np

# 例1：使用np.where()替换满足条件的元素
arr = np.array([1, 2, 3, 4, 5])
new_arr = np.where(arr < 3, 0, arr)
print(new_arr)  # 输出: [0, 0, 3, 4, 5]

# 例2：使用np.where()获取满足条件的元素的位置
arr = np.array([1, 2, 3, 4, 5])
indexes = np.where(arr > 3)
print(indexes)  # 输出: (array([3, 4]),)

# 例3：使用np.where()替换多个条件
arr = np.array([1, 2, 3, 4, 5])
new_arr = np.where((arr < 3) | (arr > 4), 0, arr)
print(new_arr)  # 输出: [0, 0, 3, 0, 5]

# 创建一个示例数组
arr = np.array([1, 2, 3, 4, 5])
arr1 = np.array([1, 2, 3, 4, 5])


new_arr = np.where(arr > 2, arr1+1, arr1)
print(new_arr)