Python 实现取多维数组第n维的前几位
更新时间：2019年11月26日 15:52:09 作者：Kenn7
今天小编就为大家分享一篇Python 实现取多维数组第n维的前几位，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
现在我们有一个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)
其实更简单的做法是：
print("# original", input.shape)
print("# output", input[:, [0, 1, 2]].shape)
以上这篇Python 实现取多维数组第n维的前几位就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了python中使用enumerate函数遍历元素实例,这是一个比较简单的例子,需要的朋友可以参考下2014-06-06
python自动打开浏览器下载zip并提取内容写入excel
这篇文章主要给大家介绍了关于python自动打开浏览器下载zip并提取内容写入excel的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-01-01
这篇文章主要介绍了python实现的多线程端口扫描功能,结合实例形式分析了Python基于socket的端口扫描具体步骤与相关操作技巧,需要的朋友可以参考下2017-01-01
这篇文章主要介绍了Python命令行参数解析之argparse模块详解,argparse 是 Python 的一个标准库，用于命令行参数的解析，这意味着我们无需在代码中手动为变量赋值，而是可以直接在命令行中向程序传递相应的参数，再由变量去读取这些参数,需要的朋友可以参考下2023-08-08
jupyter (之前的 ipython notebook )于我的最大意义在于,让学习进程和探索进程变得可累积，正如它的原先名字中的 notebook 所暗示的那样,作为学习的记录者，方便你随时捡起学习的进度，增量式地前进2016-03-03
这篇文章主要为大家详细介绍了使用python实现名片管理系统，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2020-06-06
这篇文章主要为大家详细介绍了python中yield函数的用法，数据库，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2022-02-02
百度AI功能还是很强大的,百度AI开放平台真的是测试接口的天堂,免费接口很多,当然有量的限制,但个人使用是完全够用的,下面这篇文章主要给大家介绍了关于Python调用百度AI实现图片上文字识别功能的相关资料,需要的朋友可以参考下2021-09-09
这篇文章主要介绍了python爬虫 urllib模块url编码处理详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-08-08
这篇文章主要介绍了python tkinter组件使用详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-09-09
最新评论