np.where用法以及常见的用法示例

本文详细介绍了np.where在处理一维和二维数组时的使用方法,通过实例展示了如何利用该函数进行条件判断和数据选取。
摘要由CSDN通过智能技术生成

np.where常用的用法示例,从一维数组以及二维数组进行举例:

一、一维数组的np.where用法
import numpy as np

arr = np.arange(6)
print(arr)  # [0 1 2 3 4 5]
print(np.where(arr > 3))  # (array([4, 5], dtype=int64),) 代表大于3数值的pos
print(np.where(arr > 3, 1, -1))  # [-1 -1 -1 -1  1  1]
arr = np.array([1, 3, 4, 5, 1]
np.where是numpy库中的一个函数,用于根据条件返回一个新的数组。它可以有三种不同的用法: 1. 三个参数的用法:在这种用法中,np.where接受一个条件、一个满足条件时的返回值和一个不满足条件时的返回值。它根据条件对每个元素进行判断,并将满足条件的元素替换为满足条件的返回值,不满足条件的元素替换为不满足条件的返回值。例如,如果我们有一个数组data,我们可以使用np.where((data>=0) & (data<=2), np.ones_like(data), np.zeros_like(data))来将数组中满足条件的元素替换为1,不满足条件的元素替换为0。 2. 一个参数的用法:在这种用法中,np.where只接受一个条件作为参数。它返回一个元组,其中包含满足条件的元素的索引。例如,如果我们有一个数组a,我们可以使用np.where(a>5)来获取数组中大于5的元素的索引。 3. 代码示例用法:这是一个具体的代码示例,展示了np.where的用法。在这个示例中,我们首先生成一个随机数组x,然后使用np.where(x>0, 2, -2)来将数组中大于0的元素替换为2,小于或等于0的元素替换为-2。 综上所述,np.where是一个非常有用的函数,可以根据条件来进行数组元素的替换或索引的获取。它在数据处理和分析中经常被使用。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *2* *3* [pythonnp.where()的使用方法](https://blog.csdn.net/weixin_46713695/article/details/127340899)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 100%"] [ .reference_list ]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值