2011-12-26 5 views
0

2 개의 배열이 있습니다. 하나는 2 * 2 배열입니다.모양이 다른 수의 배열을 추가하는 방법은 무엇입니까?

a = [[1,2],[3,4]] 

다른 하나는 2 * 2 * 4 어레이입니다.

b = [[[0,0,0,0],[0,0,0,0]],[[0,0,0,0],[0,0,0,0]]] 

내가 2 * 2 * 4 배열을 갖도록 추가하려고합니다. c.

c = [[[1,0,0,0],[2,0,0,0]],[[3,0,0,0],[4,0,0,0]]] 

올바른 방법은 무엇입니까?

편집 : 이것은 하나의 단계로 합계를 할 수 있는지 여부를 확실하지

b[:,:,:1]+=a[:,:,np.newaxis] 

답변

0

작동하도록 나타납니다. 여기에 2 단계로 나와 있습니다.

c = b.copy() 
c[...,0] += a 
+0

감사합니다. 지금 제 문제를 봅니다. 내가 'b [:, :, :] 1을 b [:, :, 0]'과 같게하려고했다. – user816555

관련 문제