python提取矩阵主对角线代码

你可以使用NumPy库来提取矩阵的主对角线。下面是一个示例代码：

import numpy as np

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

# 提取主对角线
diagonal = np.diagonal(matrix)

print(diagonal)

这段代码会输出矩阵的主对角线元素：[1 5 9]。你可以根据需要修改矩阵的大小和内容。