python由已知数组快速生成新数组的方法


Posted in Python onApril 08, 2020

需求描述

在利用numpy进行数据分析时,常有的一个需求是:根据已知的数组生成新数组。这个问题又可以分为两类:

  1. 根据筛选条件生成子数组;
  2. 根据变换条件生成新数组(新数组shape与原数组相同)

下面简单总结.

生成子数组

情况1

已知数组a,以及若干筛选条件conds,要求从数组a中生成一个子数组b。

解决办法:b=a[conds]比如b=a[a>0],b=a[(a>=1)|(a<=-2)], b=a[(a>=1)&(a<=3)]

实例:如下

# 实例1.1:已知数组a,要求找出所有a>0的元素,然后生成一个新数组。
a = np.arange(-5,5,1)
print('原数组a:',a)
b = a[a>0]
print('实例1结果:',b)

# 实例1.2:已知数组a,要求找出所有a>=1或a<=-2的元素,然后生成一个新数组。
b = a[(a>=1) | (a<=-2)]
print('实例2结果:',b)

# 实例1.3:已知数组a,要求找出所有a>=1并且a<=3的元素,然后生成一个新数组。
b = a[(a>=1) & (a<=3)]
print('实例3结果:',b)

运行结果:

原数组a: [-5 -4 -3 -2 -1  0  1  2  3  4]
实例1结果: [1 2 3 4]
实例2结果: [-5 -4 -3 -2  1  2  3  4]
实例3结果: [1 2 3]

情况2

已知数组a和数组b(shape相同),以及对数组a的筛选条件conds_a。要求从数组b中生成一个子数组c,其中的元素id,与满足筛选条件的数组a的元素id一一对应。

解决办法:

c = b[conds_a],比如c = b[(a>=1) & (a<=3)]

实例:如下

a = np.arange(-5,5,1)
b = np.arange(-50,50,10)
print('数组a:',a)
print('数组b:',b)

c = b[(a>=1) & (a<=3)]
print('新数组c:', c)

运行结果:

数组a: [-5 -4 -3 -2 -1  0  1  2  3  4]
数组b: [-50 -40 -30 -20 -10   0  10  20  30  40]
新数组c: [10 20 30]

变换成新数组

已知数组a,以及若干变换条件conds,要求生成一个新数组b(与原数组shape相同)。解决办法:

方法1:np.where(where(condition, [x, y]))
使用场景:当变换条件只有两个以下时,比如实例2.1。该方法等价于:if condition x else y

方法2: np.select(condlist, choicelist, default=0)
使用场景:当变换条件有任意多个时,比如实例2.2。该方法等价于:

if condlist[0]:
  choicelist[0]
elif condilist[1]:
  choicelist[1]
...
else:
  default

方法3:np.piecewise(x, condlist, funclist, *args, **kw)

使用场景:同方法2,只不过变换条件较复杂,无法直接写出,要用函数来表示。
实例:如下

# 实例2.1:已知数组a,要求对所有a<0的元素取绝对值,对其他元素设为0,然后生成一个新数组
a = np.arange(-5,5,1)
print(a)
b = np.where(a<0, abs(a),0)
print('实例2.1结果:',b)

# 实例2.2:已知数组a,要求对所有a<0的元素取绝对值,对a=0的元素+100,对a>0的元素平方,然后生成一个新数组
b = np.select([a<0, a==0, a>0], [abs(a), a+100, a**2])
print('实例2.2结果:',b)

运行结果:

[-5 -4 -3 -2 -1  0  1  2  3  4]
实例2.1结果: [5 4 3 2 1 0 0 0 0 0]
实例2.2结果: [5 4 3 2 1 100 1 4 916]

到此这篇关于python由已知数组快速生成新数组的方法的文章就介绍到这了,更多相关python 已知数组快速生成新数组内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木

Python 相关文章推荐
python中使用正则表达式的连接符示例代码
Oct 10 Python
Python cookbook(数据结构与算法)从序列中移除重复项且保持元素间顺序不变的方法
Mar 13 Python
Python返回数组/List长度的实例
Jun 23 Python
解决python selenium3启动不了firefox的问题
Oct 13 Python
python3.6.3转化为win-exe文件发布的方法
Oct 31 Python
详解利用python+opencv识别图片中的圆形(霍夫变换)
Jul 01 Python
pandas条件组合筛选和按范围筛选的示例代码
Aug 26 Python
Python concurrent.futures模块使用实例
Dec 24 Python
pyinstaller还原python代码过程图解
Jan 08 Python
设置jupyter中DataFrame的显示限制方式
Apr 12 Python
Scrapy+Selenium自动获取cookie爬取网易云音乐个人喜爱歌单
Feb 01 Python
Sentry的安装、配置、使用教程(Sentry日志手机系统)
Jul 23 Python
TensorFlow2.X使用图片制作简单的数据集训练模型
Apr 08 #Python
解决python ThreadPoolExecutor 线程池中的异常捕获问题
Apr 08 #Python
使用Python将Exception异常错误堆栈信息写入日志文件
Apr 08 #Python
TensorFlow2.X结合OpenCV 实现手势识别功能
Apr 08 #Python
python 安装库几种方法之cmd,anaconda,pycharm详解
Apr 08 #Python
TensorFlow2.1.0最新版本安装详细教程
Apr 08 #Python
解决python多线程报错:AttributeError: Can't pickle local object问题
Apr 08 #Python
You might like
php无限极分类实现的两种解决方法
2013/04/28 PHP
深入掌握include_once与require_once的区别
2013/06/17 PHP
PHP中使用break跳出多重循环代码实例
2015/01/21 PHP
PHP记录搜索引擎蜘蛛访问网站足迹的方法
2015/04/15 PHP
浅谈PHP接入(第三方登录)QQ登录 OAuth2.0 过程中遇到的坑
2017/10/13 PHP
php实现统计二进制中1的个数算法示例
2018/01/23 PHP
thinkphp5框架实现数据库读取的数据转换成json格式示例
2019/10/10 PHP
Javascript学习笔记1 数据类型
2010/01/11 Javascript
常见效果实现之返回顶部(结合淡入、淡出、减速滚动)
2012/01/04 Javascript
jQuery阻止同类型事件小结
2013/04/19 Javascript
解析jQuery的三种bind/One/Live事件绑定使用方法
2013/12/30 Javascript
js模拟hashtable的简单实例
2014/03/06 Javascript
简单对比分析JavaScript中的apply,call与this的使用
2015/12/04 Javascript
jquery实现一个简单的表单验证实例
2016/03/30 Javascript
Angular.js回顾ng-app和ng-model使用技巧
2016/04/26 Javascript
使用Script元素发送JSONP请求的方法
2016/06/12 Javascript
利用JavaScript阻止表单提交的两种方法
2016/08/11 Javascript
微信小程序 安全包括(框架、功能模块、账户使用)详解
2017/01/16 Javascript
JavaScript登录验证基础教程
2017/11/01 Javascript
JavaScript 点击触发复制功能实例详解
2018/11/02 Javascript
使用webpack构建应用的方法步骤
2019/03/04 Javascript
js实现继承的方法及优缺点总结
2019/05/08 Javascript
[01:02:10]DOTA2上海特级锦标赛B组小组赛#2 VG VS Fnatic第一局
2016/02/26 DOTA
详解Python的Django框架中的模版继承
2015/07/16 Python
利用Python如何将数据写到CSV文件中
2018/06/05 Python
Python中正则表达式的用法总结
2019/02/22 Python
解决django服务器重启端口被占用的问题
2019/07/26 Python
Django实现文章详情页面跳转代码实例
2020/09/16 Python
html5实现完美兼容各大浏览器的播放器
2014/12/26 HTML / CSS
Agoda西班牙:全球特价酒店预订
2017/06/03 全球购物
北美主要的汽车零部件零售商:AutoShack.com
2019/02/23 全球购物
求职信的要素有哪些呢
2013/12/26 职场文书
《王二小》教学反思
2014/02/27 职场文书
葛优非诚勿扰搞笑征婚台词
2014/03/17 职场文书
民警群众路线教育实践活动对照检查材料
2014/10/04 职场文书
2015年思想品德教学工作总结
2015/07/22 职场文书