Как вы последовательно переверните каждое измерение в массиве 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). Как я могу это сделать в Python, учитывая, что X является массивом NumPy?

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