Numpy - 대각선 곱을 계산합니다.

Question

새내기 질문 : 행렬 A를 가정하면 A의 대각선 요소 곱을 계산할 수있는 함수가 있습니까?

EG :

, X (I, i)는 매트릭스의 대각선 제품

.

Answer 1

당신은 diagonal 및 prod 방법을 사용할 수

import numpy as np 
a = np.matrix([[1, 1, 1], 
      [1, 2, 3], 
      [3, 3, 3]]) 

prod_diag = a.diagonal().prod() 
print(prod_diag) 
# gives 6 as answer 

diagonal는 1 차원 배열로 행렬의 대각 요소를 반환 prod 어레이의 모든 요소의 곱을 계산한다.