python diag函数的作用是什么

numpy.diag() 函数在 NumPy 库中用于提取二维数组（矩阵）的对角线元素，或者创建一个新的对角线矩阵

1. 提取对角线元素：当传入一个二维数组时，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] 

2. 创建对角线矩阵：当传入一个一维数组时，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，可以用来指定要提取或创建的对角线的位置。正值表示上对角线，负值表示下对角线。