python – 在最后2个维度中旋转5D数组

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

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

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

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

我尝试过的：

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

rot90函数显然会旋转整个数组.

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

>>> a180 = np.rot90(a,axes=(-2,-1))
>>> a180[0,0]
array([[3,0]])

如果您的NumPy版本没有np.rot90的axes参数,则有其他选择.

一种方法是索引：

a180 = a[...,::-1,::-1]

rot90翻转阵列的前两个轴,因此要使用它,您需要移调(反转轴),旋转并再次移调.例如：

np.rot90(a.T,2).T

（编辑：安卓应用网）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!