np.where(condition)

本文介绍如何利用NumPy的np.where函数筛选数组中满足特定条件的元素,并获取其索引位置。通过实例演示了单条件及复合条件的筛选过程。

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

np.where返回满足内部条件的数列a中的索引值。

np.where(condition)[0]返回一个numpy数列。

a=np.array([3,4,5,7,8])
np.where(a>7)[0]
#output:array([4])
a=np.array([3,4,5,7,8])
b=np.array([1,2,3,4,5])
np.where((a>=5)&(b>=4))
#output:(array([3, 4]),)

 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值