np.diag(array) 中
- array是一個1維數組時，結果形成一個以一維數組為對角線元素的矩陣
- array是一個二維數組時，結果輸出數組的對角線元素
舉例：
>>> a
array([1, 2, 3])
>>> b
array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
>>> np.diag(a)
array([[1, 0, 0],
[0, 2, 0],
[0, 0, 3]])
>>> np.diag(b)
array([1, 5, 9])