python np.diag

`np.diag` 是 NumPy 库中的一个函数，可以用于创建对角矩阵或提取矩阵的对角线元素。

创建对角矩阵：

import numpy as np

# 创建一个3x3的对角矩阵，对角线元素为 [1, 2, 3]
diag_matrix = np.diag([1, 2, 3])
print(diag_matrix)

输出：

[[1 0 0]
 [0 2 0]
 [0 0 3]]

提取矩阵的对角线元素：

import numpy as np

matrix = np.array([[1, 2, 3],
                   [4, 5, 6],
                   [7, 8, 9]])

# 提取矩阵的对角线元素
diagonal = np.diag(matrix)
print(diagonal)

输出：

[1 5 9]

希望能够帮到你！如果有其他问题，请随时提问。