当前位置:  开发笔记 > 编程语言 > 正文

在最后2个维度中旋转5D阵列

如何解决《在最后2个维度中旋转5D阵列》经验,为你挑选了0个好方法。

我有一个5D数组'a',大小(3,2,2,2,2).

import numpy as np
a = np.arange(48).reshape(3,2,2,2,2)
a[0,0,0]:
array([[0, 1],
       [2, 3]])

我想要做的是将这个5D阵列旋转180度,但仅在最后两个维度中,不改变它们的位置.所以输出[0,0,0]应如下所示:

out[0,0,0]:
array([[3, 2],
       [1, 0]])

我尝试过的:

out = np.rot90(a, 2)
out[0,0,0]:
array([[40, 41],
       [42, 43]])

rot90功能显然可以旋转整个阵列.

注意:如果可能,我想避免使用for循环

推荐阅读
U友50081205_653
这个屌丝很懒,什么也没留下!
DevBox开发工具箱 | 专业的在线开发工具网站    京公网安备 11010802040832号  |  京ICP备19059560号-6
Copyright © 1998 - 2020 DevBox.CN. All Rights Reserved devBox.cn 开发工具箱 版权所有