[numpy tutorial] numpy에서 대각선 값 채우기 in python

Python :: fill diagonal in numpy

 

numpy에서 대각선 값을 채우기 위해서는 numpy.fill_diagonal() 를 활용하면 된다.

 

numpy를 np 로 import 하고, 0부터 15로 이루어진 4 by 4 테이블을 생성해 주었다.

 

import numpy as np

metrix = np.arange(16).reshape(4,4)

 

생성된 metrix 결과는 다음과 같다.

 

np.diagonal(metrix) 로 대각선 값이 무엇인지 확인할 수 있다.

 

 

 

---

대각선 값을 모두 0으로 지정하는 방법은 다음과 같다.

 

np.fill_diagonal(metrix, 0)

 

모두 한 가지 값이 아니라 각각 다른 값으로도 채워줄 수 있다.

 

대각선 값을 100, 101, 102, 103 으로 채워보았다.

 

np.fill_diagonal(metrix, [100,101, 102, 103])