当前位置: 首页 > 编程笔记 >

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

巫马曜文
2023-03-14
本文向大家介绍Python 实现取多维数组第n维的前几位,包括了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维的前几位就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持小牛知识库。

 类似资料:
  • 本文向大家介绍PHP实现判断数组是一维、二维或几维的方法,包括了PHP实现判断数组是一维、二维或几维的方法的使用技巧和注意事项,需要的朋友参考一下 本文实例讲述了PHP实现判断数组是一维、二维或几维的方法。分享给大家供大家参考,具体如下: 这里使用的自定义函数,可以判断数组是一维的,还是二维的,或是几维的数组: 验证过可以使用: 更多关于PHP相关内容感兴趣的读者可查看本站专题:《PHP数组(Ar

  • 问题内容: 我需要有一个n维字段,其中n基于构造函数的输入。但是我什至不确定这是否可行。是吗? 问题答案: 快速的解决方案:你可以用非通用近似它的的…要深,因为你需要。但是,使用快速可能会很尴尬。 另一种需要更多工作的选择可能是使用基础平面数组表示形式来实现您自己的类型,在其中您内部计算索引,并为访问器方法提供vararg参数。我不确定它是否完全可行,但可能值得一试… 粗略的示例(未经测试,没有溢

  • 本文向大家介绍python numpy 一维数组转变为多维数组的实例,包括了python numpy 一维数组转变为多维数组的实例的使用技巧和注意事项,需要的朋友参考一下 如下所示: 借助numpy库; 以上这篇python numpy 一维数组转变为多维数组的实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持呐喊教程。

  • 问题内容: 我有一个要转换为Python的Java小问题。因此,我需要一个多维数组。在Java中,它看起来像: 其他值将在bei循环中创建并写入数组。 如何实例化数组? PS:不涉及矩阵乘法… 问题答案: 您可以使用嵌套列表创建它: 如果必须动态的话,它会更加复杂,为什么不自己编写一个小类呢? 可以这样使用: 我敢肯定,可以更有效地实施它。:) 如果您需要多维数组,则可以创建一个数组并计算偏移量,

  • 本文向大家介绍使用python实现多维数据降维操作,包括了使用python实现多维数据降维操作的使用技巧和注意事项,需要的朋友参考一下 一,首先介绍下多维列表的降维 二、这种降维方法同样适用于多维迭代器的降维 iterable:可迭代的,迭代器,在Python中iterable被认为是一个对象,这个对象可以一次返回它的一个成员(也就是对象里面的元素),Python中的string,list,tup

  • 本文向大家介绍Python numpy多维数组实现原理详解,包括了Python numpy多维数组实现原理详解的使用技巧和注意事项,需要的朋友参考一下 NumPy(Numerical Python) 是 Python 语言的一个扩展程序库,支持大量的维度数组与矩阵运算,此外也针对数组运算提供大量的数学函数库。今天就针对多维数组展开来写博客numpy其一部分功能如下: 1.ndarray,是具有矢量