我有一个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循环
最佳答案
要将最后两个轴旋转180度,请将轴=( – 2,-1)传递给np.rot90：
原文链接：https://www.f2er.com/python/438467.html>>> 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