Python 科学计算 | 三维数组的转置运算

在 numpy 中可以使用 transpose 和 einsum 两种方法进行三维数组的转置.

>>> x = np.arange(12).reshape(3,2,2)
>>> x
array([[[ 0,  1],
        [ 2,  3]],

       [[ 4,  5],
        [ 6,  7]],

       [[ 8,  9],
        [10, 11]]])
>>> x.transpose(2,1,0)
array([[[ 0,  4,  8],
        [ 2,  6, 10]],

       [[ 1,  5,  9],
        [ 3,  7, 11]]])
>>> np.einsum('ijk->kji',x)
array([[[ 0,  4,  8],
        [ 2,  6, 10]],

       [[ 1,  5,  9],
        [ 3,  7, 11]]])