你如何顺序翻转NumPy数组中的每个维度？

Question

我在MATLAB中遇到了以下函数,它顺序翻转矩阵中的所有维度:

function X=flipall(X)
    for i=1:ndims(X)
        X = flipdim(X,i);
    end
end

哪里X有尺寸(M,N,P) = (24,24,100).如果X是NumPy数组,我怎么能在Python中做到这一点？

Answer 1

flipdim在MATLAB中的等价物是flip在numpy.请注意,这仅适用于1.12.0版.

因此,它很简单:

import numpy as np

def flipall(X):
    Xcopy = X.copy()
    for i in range(X.ndim):
        Xcopy = np.flip(Xcopy, i)
     return Xcopy

因此,您只需将其称为:

Xflip = flipall(X)

但是,如果您事先知道只有三个维度,则只需执行以下操作即可对操作进行硬编码:

def flipall(X):
    return X[::-1,::-1,::-1]

这会使每个尺寸一个接一个地翻转.

如果您没有版本1.12.0(感谢用户hpaulj),您可以使用slice相同的操作:

import numpy as np

def flipall(X):
    return X[[slice(None,None,-1) for _ in X.shape]]