diag作用和用法

萌萌哒的杂粮饼

diag作用和用法？

FishC

diag是Python中的一个函数，用于提取或构建对角矩阵。

使用方法如下：

1. 提取对角线元素：只需将要提取对角线元素的二维数组作为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. 构建对角矩阵：可以通过将一维数组（或列表）作为diag的参数传入，函数将返回一个以该数组为主对角线元素的二维数组。

import numpy as np

   diag_elements = [1, 2, 3]
   diag_matrix = np.diag(diag_elements)
   print(diag_matrix)
   # 输出:
   # [[1 0 0]
   #  [0 2 0]
   #  [0 0 3]]

在科学计算和数据处理中，diag函数经常用于操作对角矩阵，以及从矩阵中提取对角线元素。

以上回复来自 FishC AI 小助理，如未能正确解答您的问题，请继续追问。