A = zeros(5,5);
mask = [ 0, 1, 0, 0, 1];
A(mask, mask) = 5;

>> A = [ 0 5 0 0 5 
         0 5 0 0 5 
         0 5 0 0 5 
         0 5 0 0 5 
         0 5 0 0 5  ];

위의 matlab code를 아래와 같이 바꾸어 보았습니다.

A = np.zeros((5,5));
mask = np.array( [ 0, 1, 0, 0, 1] );
A_temp = np.zeros((1,5));
A_temp[mask.nonzero()] = 5;

For k in range(5):
    A[k, : ] =  A_temp;

더 간단하고 직관적인 방법이 있을까요?