np.where是NumPy库中的一个函数,用于根据给定的条件返回一个新的数组,其中满足条件的元素被替换为指定的值,不满足条件的元素保持不变。
np.where的语法如下: np.where(condition, x, y)
参数说明:
应用场景: np.where常用于根据条件对数组进行元素级别的操作,例如根据某个条件筛选数组中的元素,或者根据条件将数组中的元素替换为指定值。
示例代码:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
new_arr = np.where(arr > 3, 10, arr)
print(new_arr)
输出结果:
[1 2 3 10 10]
推荐的腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云