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

发布时间：2020-12-16 22:50:36 所属栏目：Python 来源：网络整理

导读：我有一个5D数组a,大小(3,2,2). import numpy as npa = 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]]) 我尝试

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

>>> 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

（编辑：李大同）

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