Python 矩阵对角元素提取：使用 NumPy diag() 函数

要提取一个矩阵的对角元素，可以使用 NumPy 库中的 diag() 函数。以下是一个示例代码：

import numpy as np

# 创建一个3x3的矩阵
matrix = np.array([[1, 2, 3],
                   [4, 5, 6],
                   [7, 8, 9]])

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

print(diagonal)

输出结果为：

[1 5 9]

在这个示例中，我们首先导入 NumPy 库。然后，创建一个 3x3 的矩阵 'matrix'。最后，使用 'np.diag()' 函数提取 'matrix' 的对角元素，并将结果保存在 'diagonal' 变量中。最后，打印 'diagonal' 的值。