numpy中实现ndarray数组返回符合特定条件的索引方法


Posted in Python onApril 17, 2018

在numpy的ndarray类型中,似乎没有直接返回特定索引的方法,我只找到了where函数,但是where函数对于寻找某个特定值对应的索引很有用,对于返回一定区间内值的索引不是很有效,至少我没有弄明白应该如何操作尴尬。下面先说一下where函数的用法吧。

(1)where函数的使用场景:

例如现在我生成了一个数组:

import numpy as np 
arr=np.array([1,1,1,134,45,3,46,45,65,3,23424,234,12,12,3,546,1,2])

现在arr是一个含有18个元素的ndarray类型的数组,后面就用数组来称呼它吧。假如我想返回数组中所有元素值为3所对应的索引位置,那么通过where函数可以很轻松地完成。

print np.where(arr==3)

它会返回一个包含所有值为3对应的索引位置的元组,如下图:

numpy中实现ndarray数组返回符合特定条件的索引方法

可以看到在索引为5、9、14的地方所对应的元素值为3。通过这样的方式可以轻松达到目的。但是对于一定变化范围内的索引这样就不行了。下面的方法是自己想出来的折衷办法,比较笨,高手清喷。

(2)通过一个辅助的数组来解决一定范围内值索引的查找

我们建一个标识元素索引的数组,然后通过它来显示符合条件的元素对应的索引。还是刚才的数组,假如我现在要返回元素值在3到100之间值的索引。我可以生成一个和arr相同大小的数组,然后首先通过一次筛选找到大于3的元素对应索引的数组,继而在其中再次筛选,最终得到想要的结果。代码如下:

b=np.arange(len(arr))#生成和arr相同长度的数组
c=b[arr>3]#c存放的就是arr中大于3的元素对应的索引 
#最后通过遍历c数组,选择3到100之间的值打印出来 
for i in range(len(c)): 
 if arr[c[i]]<100: 
  print c[i],

下面看一下执行的效果:

numpy中实现ndarray数组返回符合特定条件的索引方法

可以看到程序将所有介于3和100之间的元素对应的索引值打了出来,如果想同时获得索引值和相应的元素,只需将上面的“print c[i]”, 替换为 “print c[i],arr[c[i]]”即可。

当然这种方法也适用于选择某个特定值对应的索引,例如我想找到所有3对应的位置,可以用print b[arr==3]即可打印出所有值为3的元素所对应的索引。其实不论怎么做,都是用到了数组进行关系运算后生成布尔数组,然后让数组中为True的地方进行显示。

当然还可以通过两次进行筛选,分别筛选出大于3的元素和小于100的元素对应的索引数组,然后两个数组做交集处理,在numpy中有一个intersect1d函数可以进行这样的运算,但是仍然麻烦。目前只能想到这些办法,不知道有哪位高手有更好的方法,欢迎大家一起交流一下。

以上这篇numpy中实现ndarray数组返回符合特定条件的索引方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python ORM框架SQLAlchemy学习笔记之数据查询实例
Jun 10 Python
Python中urllib2模块的8个使用细节分享
Jan 01 Python
Python中设置变量作为默认值时容易遇到的错误
Apr 03 Python
详解在Python程序中使用Cookie的教程
Apr 30 Python
详解Django之auth模块(用户认证)
Apr 17 Python
python-docx修改已存在的Word文档的表格的字体格式方法
May 08 Python
详解用Python实现自动化监控远程服务器
May 18 Python
Python netmiko模块的使用
Feb 14 Python
Python的PIL库中getpixel方法的使用
Apr 09 Python
基于python实现查询ip地址来源
Jun 02 Python
详解scrapy内置中间件的顺序
Sep 28 Python
python statsmodel的使用
Dec 21 Python
python中找出numpy array数组的最值及其索引方法
Apr 17 #Python
详谈Numpy中数组重塑、合并与拆分方法
Apr 17 #Python
基于numpy.random.randn()与rand()的区别详解
Apr 17 #Python
Python实现求解括号匹配问题的方法
Apr 17 #Python
django 通过ajax完成邮箱用户注册、激活账号的方法
Apr 17 #Python
对numpy中array和asarray的区别详解
Apr 17 #Python
django 开发忘记密码通过邮箱找回功能示例
Apr 17 #Python
You might like
用PHP产生动态的影像图
2006/10/09 PHP
PHP中is_file不能替代file_exists的理由
2014/03/04 PHP
9条PHP编程小知识及易犯的小错误
2015/01/22 PHP
PHP+Mysql+jQuery查询和列表框选择操作实例讲解
2015/10/22 PHP
php版微信自动登录并获取昵称的方法
2016/09/23 PHP
prototype.js的Ajax对象
2006/09/23 Javascript
改进:论坛UBB代码自动插入方式
2006/12/22 Javascript
jQuery each()方法的使用方法
2010/03/18 Javascript
基于jQuery的模仿新浪微博时间的组件
2011/10/04 Javascript
THREE.JS入门教程(1)THREE.JS使用前了解
2013/01/24 Javascript
给应用部分的js代码设定一个统一的入口
2014/06/15 Javascript
更快的异步执行(setTimeout多浏览器)
2014/08/12 Javascript
node.js中的events.emitter.once方法使用说明
2014/12/10 Javascript
javascript实现的字符串与十六进制表示字符串相互转换方法
2015/07/17 Javascript
理解javascript中的Function.prototype.bind的方法
2017/02/03 Javascript
基于AngularJS拖拽插件ngDraggable.js实现拖拽排序功能
2019/04/02 Javascript
JS中注入eval, Function等系统函数截获动态代码
2019/04/03 Javascript
新手如何快速理解js异步编程
2019/06/24 Javascript
Vue data的数据响应式到底是如何实现的
2020/02/11 Javascript
[00:35]2016完美“圣”典风云人物:冷冷宣传片
2016/12/08 DOTA
Python保存MongoDB上的文件到本地的方法
2016/03/16 Python
python 打印直角三角形,等边三角形,菱形,正方形的代码
2017/11/21 Python
使用Python开发SQLite代理服务器的方法
2018/12/07 Python
Python列表list排列组合操作示例
2018/12/18 Python
python pexpect ssh 远程登录服务器的方法
2019/02/14 Python
聊聊python里如何用Borg pattern实现的单例模式
2019/06/06 Python
Python计算机视觉里的IOU计算实例
2020/01/17 Python
如何理解Python中的变量
2020/06/01 Python
python+flask编写一个简单的登录接口
2020/11/13 Python
Python爬虫回测股票的实例讲解
2021/01/22 Python
《大江保卫战》教学反思
2014/04/11 职场文书
安全生产先进个人事迹材料
2014/12/30 职场文书
死亡赔偿协议书
2015/01/28 职场文书
2015年个人审计工作总结
2015/04/07 职场文书
Django如何与Ajax交互
2021/04/29 Python
Nginx使用Lua模块实现WAF的原理解析
2021/09/04 Servers