如果矩陣可對角化,則回傳對角元素
Example Code

• import numpy as np
• martix0 = np.arange(9).reshape((3,3))
• print('diagonalized matrix with diag function', np.diag(martix0))
• print('diagonalized matrix with diagonal function', martix0.diagonal())

Note and Comments

• import numpy as np: 讀入numpy函式庫並命名為np
• np.arange(l).reshape((m,n)) 生成一個有九個元素的陣列, 並重新將此陣列重新分割爲3*3的矩陣
• np.diag(martix0) 對角化martix0, 並回傳對角元素
• martix0.diagonal() 對角化martix0, 並回傳對角元素

result