diag()
函數在 Python 中通常與 NumPy 庫一起使用,它用于從給定的數組中提取對角線元素或創建一個對角線矩陣
diag()
函數時,它會返回該矩陣的主對角線元素。例如:import numpy as np
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
diagonal_elements = np.diag(matrix)
print(diagonal_elements) # 輸出:[1 5 9]
diag()
函數時,它會創建一個以該向量為對角線元素的方陣。例如:import numpy as np
vector = np.array([1, 2, 3])
diagonal_matrix = np.diag(vector)
print(diagonal_matrix)
# 輸出:
# [[1 0 0]
# [0 2 0]
# [0 0 3]]
在調試過程中,diag()
函數可以幫助您檢查和操作矩陣的對角線元素,以便更好地理解代碼的行為和性能。