첨부 실행 코드는 나눔고딕코딩 폰트를 사용합니다.
728x90
반응형
728x170

■ ndarray 클래스의 diag 함수를 사용해 대각선 요소의 값만 추출한 배열을 구하는 방법을 보여준다.

 

▶ 예제 코드 (PY)

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]
"""
728x90
반응형
그리드형(광고전용)
Posted by icodebroker
TAG , ,

댓글을 달아 주세요