Python 实现取多维数组第n维的前几位


Posted in Python onNovember 26, 2019

现在我们有一个shape为(7352, 9, 128, 1)的numpy数组。

想要取出第2维的前三个数据,构成新数组(7352, 3, 128, 1)

我的思想是:将第2维数据转置(transpose)到第一维,再用切片(slice)取出前三个数据,再转置回去:

print("# original", input.shape)
input_transpose = input.transpose((1, 0, 2, 3))
print("# transpose", input_transpose.shape)
input_slice = input_transpose[0:3]
print("# slice", input_slice.shape)
output = input_slice.transpose((1, 0, 2, 3))
print("# output", output.shape)

Python 实现取多维数组第n维的前几位

其实更简单的做法是:

print("# original", input.shape)
print("# output", input[:, [0, 1, 2]].shape)

Python 实现取多维数组第n维的前几位

以上这篇Python 实现取多维数组第n维的前几位就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python局部赋值的规则
Mar 07 Python
从零学python系列之新版本导入httplib模块报ImportError解决方案
May 23 Python
编写同时兼容Python2.x与Python3.x版本的代码的几个示例
Mar 30 Python
Python循环语句中else的用法总结
Sep 11 Python
python之super的使用小结
Aug 13 Python
使用Python函数进行模块化的实现
Nov 15 Python
Pytorch 中retain_graph的用法详解
Jan 07 Python
keras获得model中某一层的某一个Tensor的输出维度教程
Jan 24 Python
python实现图像拼接
Mar 05 Python
Pytorch框架实现mnist手写库识别(与tensorflow对比)
Jul 20 Python
python IP地址转整数
Nov 20 Python
Python虚拟环境virtualenv是如何使用的
Jun 20 Python
python获取array中指定元素的示例
Nov 26 #Python
Flask之pipenv虚拟环境的实现
Nov 26 #Python
Python 根据数据模板创建shapefile的实现
Nov 26 #Python
Python实现直播推流效果
Nov 26 #Python
Python利用matplotlib绘制约数个数统计图示例
Nov 26 #Python
创建Shapefile文件并写入数据的例子
Nov 26 #Python
python使用opencv在Windows下调用摄像头实现解析
Nov 26 #Python
You might like
桌面中心(二)数据库写入
2006/10/09 PHP
php导入导出excel实例
2013/10/25 PHP
实例讲解php数据访问
2016/05/09 PHP
php使用gd2绘制基本图形示例(直线、圆、正方形)
2017/02/15 PHP
PHP终止脚本运行三种实现方法详解
2020/09/01 PHP
Aster vs Newbee BO3 第二场2.18
2021/03/10 DOTA
图片按比例缩放函数
2006/06/26 Javascript
JavaScript高级程序设计 阅读笔记(四) ECMAScript中的类型转换
2012/02/27 Javascript
用jQuery实现的智能隐藏、滑动效果的返回顶部代码
2014/03/18 Javascript
详谈javascript中的cookie
2015/06/03 Javascript
JQuery工具函数汇总
2015/06/15 Javascript
JavaScript取得WEB安全颜色列表的方法
2015/07/14 Javascript
浅谈$(document)和$(window)的区别
2015/07/15 Javascript
JavaScript的面向对象编程基础
2015/08/13 Javascript
盘点javascript 正则表达式中 中括号的【坑】
2016/03/16 Javascript
Angular中$broadcast和$emit的使用方法详解
2017/05/22 Javascript
Vue结合SignalR实现前后端实时消息同步
2017/09/19 Javascript
在Vue组件中获取全局的点击事件方法
2018/09/06 Javascript
利用JS动态生成隔行换色HTML表格的两种方法
2018/10/09 Javascript
Vuex mutitons和actions初使用详解
2019/03/04 Javascript
理理Vue细节(推荐)
2019/04/16 Javascript
Vue scoped及deep使用方法解析
2020/08/01 Javascript
微信小程序canvas实现签名功能
2021/01/19 Javascript
Python中列表的一些基本操作知识汇总
2015/05/20 Python
Python向MySQL批量插数据的实例讲解
2018/03/31 Python
Python中extend和append的区别讲解
2019/01/24 Python
Python HTML解析器BeautifulSoup用法实例详解【爬虫解析器】
2019/04/05 Python
Python数据结构dict常用操作代码实例
2020/03/12 Python
荟萃全球保健品:维他购
2018/05/09 全球购物
如何在Shell脚本中使用函数
2015/09/06 面试题
初一学生期末评语
2014/04/24 职场文书
服务承诺口号
2014/05/22 职场文书
领导干部民主生活会自我剖析材料范文
2014/09/20 职场文书
2015年世界粮食日演讲稿
2015/03/20 职场文书
Oracle11g R2 安装教程完整版
2021/06/04 Oracle
HTML怎么设置下划线?html文字加下划线方法
2021/12/06 HTML / CSS