3 차원 어레이의 평면에 넘치는 합계, 스칼라 반환

Question

나는 MATLAB에서 Numpy로 전환하면서 성장 통증을 느끼고 있습니다.

저는 3 차원 배열을 가지고 있다고 말하면 3x3x3이고 각 평면의 스칼라 합이 필요합니다. MATLAB에서 , 내가 사용합니다 :

sum_vec = sum(3dArray,3); 

TIA 세계 은행

편집 : 내 MATLAB 코드에 대해 잘못이었다. Matlab은 희미하게 만 벡터화되므로 루프가 필요합니다. 그래서 numpy는 더 우아한 것으로 밝혀진다. .. 시원한.

MATLAB 
for i = 1:3 
    sum_vec(i) = sum(sum(3dArray(:,:,i)); 
end 

Answer 1

당신은 np.sum에 axis 키워드를 사용해야합니다. 다른 많은 numpy 함수와 마찬가지로 axis을 사용하면 특정 축을 따라 작업을 수행 할 수 있습니다.

import numpy as np 
sum_vec = np.sum(3dArray, axis=-1) 

을 그리고 당신은 모든 배열 조각에 마지막 차원에 따라 합계에 해당하는 결과 2 차원 배열을 얻을 것이다 : 당신은 배열의 마지막 차원에 따라 합계를 예를 들어, 당신은 할 것 3dArray[i, k, :].

UPDATE

내가 당신이 원하는 정확히 이해하지 못했다. 당신은 2 차원 (평면)을 합산하려고합니다. 이 경우 두 개의 합계를 사용할 수 있습니다. 예를 들어, 첫 번째 두 가지 차원에 걸쳐 합산 :

sum_vec = np.sum(np.sum(3dArray, axis=0), axis=0) 

Answer 2

당신이 할 수있는

sum_vec = np.array([plane.sum() for plane in cube]) 

또는 cube이 3 차원 배열을 단순히

sum_vec = cube.sum(-1).sum(-1) 

. 플레인의 방향에 따라 -1 (또는 2) 대신 0 또는 1을 지정할 수 있습니다. 후자의 버전은 파이썬 루프를 사용하지 않기 때문에 더 좋으며 보통 numpy을 사용할 때 성능을 향상시키는 데 도움이됩니다.

Answer 3

대신 두 번 같은 합계 함수를 적용하는 경우에는 재 형성된 어레이의 합을 수행 할 수있다 : 번만 합계 때문에

a = np.random.rand(10, 10, 10) # 3D array 
b = a.view() 
b.shape = (a.shape[0], -1) 
c = np.sum(b, axis=1) 

위 빠를 것이다.

Answer 4

sumvec= np.sum(3DArray, axis=2) 

또는이뿐만 아니라

sumvec=3DArray.sum(2) 

파이썬을 기억하기 때문에 축 = 2는 3 차원을 나타내는 0에서 시작한다.

https://docs.scipy.org/doc/numpy-1.13.0/reference/generated/numpy.sum.html