두 행렬의 모든 열 쌍 추가하기

Question

두 개의 행렬 A와 B에 같은 양의 열이 있지만 일반적으로 행 수가 동일하지 않습니다. 나는에 대한 루프와 행렬 A와 B 본래의 구현에서 모든 열 쌍을 추가 할 것은 이것이다 :

이

import numpy as np 
ncol = 3 
nrow_A = 5 
nrow_B = 10 
A = np.ones((nrow_A,ncol)) 
B = np.zeros((nrow_B,ncol)) 

C = np.empty((nrow_A*nrow_B,ncol)) 
k = 0 
for i in range(nrow_A): 
    for j in range(nrow_B): 
     C[k,:] = A[i,:]+B[j,:] 
     k += 1 

이 예에서 것들로 가득 찬 50 * 3 행렬을 반환합니다. 이것은 한 줄의 코드로 가능해야한다고 생각합니다. 이것을 어떻게 할 수 있습니까?

Answer 1

당신이 약간 A을 바꿀 경우, 두 행렬을 함께 추가 항목을 연결할 수 있습니다 결과 배열의 첫 번째 축에서 :

>>> np.concatenate(A[:,np.newaxis,:] + B) 
array([[ 1., 1., 1.], 
     [ 1., 1., 1.], 
     [ 1., 1., 1.], 
     [ 1., 1., 1.], 
     [ 1., 1., 1.], 
     [ 1., 1., 1.], 
     [ 1., 1., 1.], 
     ... 

(A[:,np.newaxis,:] + B).reshape(-1, 3) 

Answer 2

꽤 아니지만 행은 같은 일을 수행하는 방송을 사용

(A[:, np.newaxis, :] + B[np.newaxis, :, :]).reshape(nrow_A * nrow_B, ncol) 

예를

import numpy as np 
ncol = 3 
nrow_A = 5 
nrow_B = 10 
A = np.random.random_sample((nrow_A,ncol)) 
B = np.random.random_sample((nrow_B,ncol)) 

C = np.empty((nrow_A*nrow_B,ncol)) 
k = 0 
for i in range(nrow_A): 
    for j in range(nrow_B): 
     C[k,:] = A[i,:]+B[j,:] 
     k += 1 

D = (A[:, np.newaxis, :] + B[np.newaxis, :, :]).reshape(nrow_A * nrow_B, ncol) 

print(np.allclose(C, D))