我有一个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循环