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 相关文章推荐
Python2.5/2.6实用教程 入门基础篇
Nov 29 Python
python中关于时间和日期函数的常用计算总结(time和datatime)
Mar 08 Python
Python SQLite3数据库操作类分享
Jun 10 Python
对Python发送带header的http请求方法详解
Jan 02 Python
selenium+python自动化测试之使用webdriver操作浏览器的方法
Jan 23 Python
Python面向对象程序设计类变量与成员变量、类方法与成员方法用法分析
Apr 12 Python
python调用pyaudio使用麦克风录制wav声音文件的教程
Jun 26 Python
python画图把时间作为横坐标的方法
Jul 07 Python
Django中自定义查询对象的具体使用
Oct 13 Python
python hash每次调用结果不同的原因
Nov 21 Python
解决windows上安装tensorflow时报错,“DLL load failed: 找不到指定的模块”的问题
May 20 Python
Python还能这么玩之只用30行代码从excel提取个人值班表
Jun 05 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
Ajax PHP分页演示
2007/01/02 PHP
js调用图片隐藏&amp;显示实现代码
2013/09/13 Javascript
jquery 中的each()跳出循环的语句
2014/05/23 Javascript
js删除数组元素、清空数组的简单方法(必看)
2016/07/27 Javascript
用Angular实时获取本地Localstorage数据,实现一个模拟后台数据登入的效果
2016/11/09 Javascript
JS制作类似选项卡切换的年历
2016/12/03 Javascript
jquery dataview数据视图插件使用方法
2016/12/23 Javascript
微信小程序 弹幕功能简单实例
2017/02/14 Javascript
详解JavaScript 中getElementsByName在IE中的注意事项
2017/02/21 Javascript
jquery mobile实现可折叠的导航按钮
2017/03/11 Javascript
React进阶学习之组件的解耦之道
2017/08/07 Javascript
JS继承与闭包及JS实现继承的三种方式
2017/10/15 Javascript
浅谈ajax请求不同页面的微信JSSDK问题
2018/02/26 Javascript
深入理解react-router 路由的实现原理
2018/09/26 Javascript
vue实现局部刷新的实现示例
2019/04/16 Javascript
Vue CLI3基础学习之pages构建多页应用
2019/06/02 Javascript
vue使用video.js进行视频播放功能
2019/07/18 Javascript
python分析网页上所有超链接的方法
2015/05/08 Python
微信跳一跳游戏python脚本
2020/04/01 Python
python3实现zabbix告警推送钉钉的示例
2019/02/20 Python
Pycharm创建项目时如何自动添加头部信息
2019/11/14 Python
Matplotlib使用Cursor实现UI定位的示例代码
2020/03/12 Python
使用python+poco+夜神模拟器进行自动化测试实例
2020/04/23 Python
详解pyinstaller生成exe的闪退问题解决方案
2020/06/19 Python
玩具反斗城西班牙网上商城:ToysRUs西班牙
2017/01/19 全球购物
英国知名美妆护肤在线商城:Zest Beauty
2018/04/24 全球购物
Omio葡萄牙:全欧洲低价大巴、火车和航班搜索和比价
2019/02/09 全球购物
世界上最好的足球商店:Unisport
2019/03/02 全球购物
Under Armour安德玛意大利官网:美国高端运动科技品牌
2020/01/16 全球购物
C语言中break与continue的区别
2012/07/12 面试题
公司中层干部的自我评价分享
2014/03/01 职场文书
企业委托书范本
2014/09/13 职场文书
2014年小学校长工作总结
2014/12/08 职场文书
储备店长岗位职责
2015/04/14 职场文书
Golang之sync.Pool使用详解
2021/05/06 Golang
服务器SVN搭建图文安装过程
2022/06/21 Servers