[파이썬] 문제 : 행렬 대각선(／가 아닌 ＼방향) 원소 곱셈

행렬에서 표시한 원소만 더하는 파이썬 코드 작성

import numpy as np

# 행렬 데이터
data = np.array([[1,2,3],
                 [4,5,6],
                 [7,8,9]])

# 행렬의 뒤집기
data2 = np.flip(data, axis=1)
print('뒤집힌 행렬\n',data2)

# 대각선 원소 찾기
v = np.diag(data2,k=0)
print('대각선 원소\n',v)

print(np.prod(v))   # 곱셈
print(np.sum(v))    # 덧셈