numpy ndarray 按条件筛选数组,关联筛选的例子
作者:y小川 时间:2021-04-12 04:58:09
最近的项目中大量涉及数据的预处理工作,对于ndarray的使用非常频繁。其中ndarray如何进行数值筛选,总结了几种方法。
1.按某些固定值筛选
如下面这段代码从,ndarray中可以筛选出数值等于3的子数组和其在原数组中的索引位置。
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])
print(np.where(arr==3))
print(arr[np.where(arr == 3)])
Output:
(array([ 5, 9, 14], dtype=int32),)
[3 3 3]
2.按多个固定值筛选
按上述方法筛选多个固定值也是可行的,将不同条件用括号括起来,之间打 | 即可。
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])
print(np.where((arr == 3) | (arr == 1)))
print(arr[np.where((arr == 3) | (arr == 1))])
Output:
(array([ 0, 1, 2, 5, 9, 14, 16], dtype=int32),)
[1 1 1 3 3 3 1]
3.按范围筛选
除了按固定值,我们还可以按一定的范围进行筛选
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])
print(np.where(arr > 3))
print(arr[np.where(arr > 3)])
Output:
(array([ 3, 4, 6, 7, 8, 10, 11, 12, 13, 15], dtype=int32),)
[ 134 45 46 45 65 23424 234 12 12 546]
如果不需要index,还可以有更快的方法
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])
print(arr[arr > 3])
Output:
[ 134 45 46 45 65 23424 234 12 12 546]
那为什么还需要用np.where呢?因为索引可以满足不同数组间的筛选,比如有a, b 两个数组我们需要筛选出所有a == 0的b时即可使用np.where函数。
import numpy as np
a = np.array([0, 0, 1, 1, 0, 1, 1, 0, 1])
b = np.arange(len(a))
print(a)
print(b)
print(b[np.where(a == 0)])
Output:
[0 0 1 1 0 1 1 0 1]
[0 1 2 3 4 5 6 7 8]
[0 1 4 7]
来源:https://blog.csdn.net/blackyuanc/article/details/77948703
标签:numpy,ndarray,筛选,数组,关联
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
asp分类算法要解决的问题
2009-09-10 16:49:00
发一个数字拼图网页游戏
2008-10-12 10:02:00
asp如何做一个密码“生成器”?
2010-07-12 18:51:00
如何利用数据库内容建立一个下拉式列表?
2010-01-01 15:46:00
网站程序员如何应对web标准
2007-05-11 16:52:00
![](https://img.aspxhome.com/file/uploadpic/20075/2007511165222957.gif)
wap开发 完整的WML文档结构详解
2008-05-21 13:39:00
python字符串连接的N种方式总结
2023-10-12 08:47:44
uniapp路由uni-simple-router实例详解
2023-09-24 21:27:04
mysql基础教程:安装与操作
2009-07-30 08:18:00
linux下安装php扩展memcache的方法
2023-11-21 19:49:02
![](https://img.aspxhome.com/file/2023/4/112714_0s.jpg)
Python中如何导入类示例详解
2023-05-09 08:35:58
ajax Google PageRank3(PR值)查询源代码
2007-11-04 13:31:00
![](https://img.aspxhome.com/file/UploadPic/200711/4/2007114142753675s.gif)
mysql导入导出命令
2011-07-04 11:28:50
ASP开发网页牢记注意事项
2013-06-28 16:20:30
Python实现邮件发送的详细设置方法(遇到问题)
2023-10-05 23:03:00
![](https://img.aspxhome.com/file/2023/9/63539_0s.png)
将SQL 2000日志迁移到SQL Server 2008
2009-03-25 16:20:00
python执行scp命令拷贝文件及文件夹到远程主机的目录方法
2023-07-10 09:12:19
十六则Dreamweaver使用快技法
2009-07-05 18:55:00
用asp编写文档搜索页面
2008-01-13 07:04:00
Python 不设计 do-while 循环结构的理由
2021-08-04 11:55:19
![](https://img.aspxhome.com/file/2023/9/68299_0s.jpg)