2D numpy 배열에서 3D 스타일 슬라이싱 사용

Question

나는 numpy 배열 (A)을 입력으로 사용하는 함수가 있습니다. 이 배열은 수학적 계산에 따라 2 차원 또는 3 차원 배열 일 수 있습니다. 배열이 2D 일 때를 제외하고는 임의의 숫자가 될 수있는 정수 m이 있습니다. m의 값은 항상 0이됩니다. A의 silce를 다른 함수에 전달하려고합니다. A는 3D 또는 2D가 될 수 있으므로 3D 스타일 슬라이싱을 시도했습니다. A는 2D

IndexError: too many indices for array 

나는 그것이 MATLAB에서 일어나는처럼 A는, 2D의 경우 Fun2에하는 전체 2D 배열을 전달 할 때

def fun(A): 
    ... some code 
    ans = fun2(A[:,:,m]) #The value of m is 0 if A is 2D 

이

은 나에게 IndexError을 제공합니다. 파이썬으로 어떻게 할 수 있습니까? 내가 사용하는 파이썬 좋은 설정은 단순히 같은, 마지막 축을 따라 m 번째 인덱스를 슬라이스 한 후 우리는 힘이 3D을 할 수 np.atleast_3d를 사용하고 같은 2

Answer 1

보인다 -

np.atleast_3d(A)[...,m] # Or np.atleast_3d(A)[:,:,m] 

여전히 어레이에 대한 뷰이므로 효율성이 떨어집니다!

In [160]: A = np.random.randint(11,99,(4,5)) 

In [161]: np.atleast_3d(A)[...,0] 
Out[161]: 
array([[13, 84, 38, 15, 26], 
     [64, 91, 29, 11, 48], 
     [25, 66, 77, 14, 87], 
     [59, 96, 98, 30, 88]]) 

In [162]: A 
Out[162]: 
array([[13, 84, 38, 15, 26], 
     [64, 91, 29, 11, 48], 
     [25, 66, 77, 14, 87], 
     [59, 96, 98, 30, 88]]) 

2) 3D : (순서 = F) 필요에 따라

가

In [163]: A = np.random.randint(11,99,(4,3,5)) 

In [164]: np.atleast_3d(A)[...,1] 
Out[164]: 
array([[34, 81, 66], 
     [56, 20, 25], 
     [45, 36, 64], 
     [82, 64, 31]]) 

In [165]: A[:,:,1] 
Out[165]: 
array([[34, 81, 66], 
     [56, 20, 25], 
     [45, 36, 64], 
     [82, 64, 31]])