본문 바로가기
대학원 공부/programming language

Numpy : np.diag (대각선 요소의 값만 추출한 배열 구하기)

by 월곡동로봇팔 2020. 2. 21.

np.diag

import numpy as np

sourceList = [(1, 2, 3), (4, 5, 6), (7, 8, 9)]

sourceNDArray = np.array(sourceList)
targetNDArray = np.diag(sourceNDArray)

print(targetNDArray)

[결과]
[1 5 9]

출처: https://icodebroker.tistory.com/5277

 

[PYTHON/NUMPY] diag 함수 : 대각선 요소의 값만 추출한 배열 구하기

■ diag 함수 : 대각선 요소의 값만 추출한 배열 구하기 ---------------------------------------------------------------------------------------------------- import numpy as np sourceList = [(1, 2, 3),..

icodebroker.tistory.com

 

댓글