numpy.diag() 函數在 NumPy 庫中，主要用于提取矩陣對角線上的元素或者創建一個對角矩陣
numpy.diag() 函數會返回該矩陣主對角線上的元素。可以通過設置 k 參數來獲取其他對角線上的元素。import numpy as np
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
main_diagonal = np.diag(matrix) # 結果為 array([1, 5, 9])
numpy.diag() 函數會根據該數組創建一個對角矩陣。可以通過設置 k 參數來創建一個非主對角線的對角矩陣。import numpy as np
vector = np.array([1, 2, 3])
diag_matrix = np.diag(vector) # 結果為 array([[1, 0, 0], [0, 2, 0], [0, 0, 3]])
numpy.diag() 函數可以用于將一維數組轉換為對角矩陣，或者將二維矩陣的對角線元素提取到一維數組。這種轉換在線性代數和矩陣運算中非常有用。import numpy as np
# 將向量轉換為對角矩陣
vector = np.array([1, 2, 3])
diag_matrix = np.diag(vector) # 結果為 array([[1, 0, 0], [0, 2, 0], [0, 0, 3]])
# 提取矩陣的對角線元素
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
main_diagonal = np.diag(matrix) # 結果為 array([1, 5, 9])
總之，numpy.diag() 函數在 NumPy 庫中具有廣泛的應用，包括提取和創建對角線元素、處理向量和矩陣之間的轉換等。
