你如何顺序翻转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]]