当前位置: 首页 > 面试题库 >

numpy列出第二个轴

汤嘉平
2023-03-14
问题内容

我想基于内轴拆分nd numpy数组。

我有各种各样的形状 (6,150,29,29,29,1)

我想要一个数组列表- [150 arrays of shape (6,29,29,29,1)]

我使用了list(a),但这给了我0轴的列表。


问题答案:

arr.transpose(1,0,2,3,4,5)np.swapaxes(arr,0,1)将150维放在首位。然后就可以使用了list

或者您可以使用列表理解

[a[:,i] for i in range(150)]

移调效果更好

In [28]: timeit list(arr.transpose(1,0,2,3,4,5))
47.7 µs ± 47.1 ns per loop (mean ± std. dev. of 7 runs, 10000 loops each)
In [29]: timeit [arr[:,i] for i in range(150)]
88.7 µs ± 22.2 ns per loop (mean ± std. dev. of 7 runs, 10000 loops each)
In [32]: timeit list(np.swapaxes(arr,0,1))
49.2 µs ± 51.1 ns per loop (mean ± std. dev. of 7 runs, 10000 loops each)


 类似资料:
  • 问题内容: 给定一个具有两列的文件,以标准空白隔开 我如何输出第二列 问题答案: 由于示例数据的最后一行没有第一列,因此您必须将其解析为固定宽度的列:

  • 问题内容: 假设我有以下形式的numpy数组: 我想找到第一个索引(对于每一列)的索引,其中值非零。 因此,在这种情况下,我希望返回以下内容: 我该怎么办? 问题答案: 首次出现的指标 使用沿轴(零轴这里列)非零的面具拿到第一的指标(真实值) 扩展到涵盖通用轴说明符,并且在沿着该轴找不到元素的非零的情况下,我们将有一个类似的实现- 请注意,由于所有值都会返回,因此如果需要的话,我们将直接使用生成最

  • 问题内容: 我有这样的结构: 我想使用内置方法(在一行中)对整数()求和。 有任何想法吗? 问题答案: 会工作。

  • 部署项目后,当客户机第一次向TestServlet发送请求时,server会创建TestServlet对象,然后调用第一个init()方法(init(ServletConfig config))。然后JVM检查TestServlet中的第一个init()方法,因为它不可用,然后它检查超级类HttpServlet,第一个init方法也不可用,然后JVM检查HttpServlet的超级类,即Gener

  • 问题内容: 我有两个选择: 如果我首先选择“水果”,该如何使用jQuery?第二个选择仅向我显示水果- 香蕉,苹果,橙子。如果我在第一个选择中选择“鸟”,则第二个选择仅显示“鸟”-老鹰,鹰。等等… 我试图用这段jQuery代码来做到这一点: 不幸的是,它几乎删除了所有内容,而且我不知道如何恢复一些选择。我也阅读了一些有关克隆的内容,但是在这种情况下我不知道如何使用它。 问题答案: $(“#sele

  • 问题内容: 我有一个二进制数组,我想将其转换为整数列表,其中每个int是该数组的一行。 例如: 我想转换成。 问题答案: 我曾经在这里问过类似的问题。这是我的回答,适合您的问题: