numpy.diag() 函數在 NumPy 庫中用于提取二維數組（矩陣）的對角線元素，或者創建一個新的對角線矩陣
numpy.diag() 函數會返回該數組的主對角線元素。例如：import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
diag_elements = np.diag(arr)
print(diag_elements) # 輸出：[1 5 9]
numpy.diag() 函數會根據該數組的元素創建一個新的對角線矩陣。例如：import numpy as np
arr = np.array([1, 2, 3])
diag_matrix = np.diag(arr)
print(diag_matrix)
# 輸出：
# [[1 0 0]
# [0 2 0]
# [0 0 3]]
注意：numpy.diag() 函數還有一些其他參數，如 k，可以用來指定要提取或創建的對角線的位置。正值表示上對角線，負值表示下對角線。
