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

Question

我有一个 5 维数组“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 循环

Answer 1

要将最后两个轴旋转 180 度，将 axes=(-2, -1) 传递给 np.rot90:

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

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

一种方法是使用索引:

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

rot90 翻转数组的前 两个轴，因此要使用它，您需要转置(反转轴)、旋转并再次转回.例如:

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