概述
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
总结
以上是编程之家为你收集整理的python – 在最后2个维度中旋转5D数组全部内容,希望文章能够帮你解决python – 在最后2个维度中旋转5D数组所遇到的程序开发问题。
如果您也喜欢它,动动您的小指点个赞吧