python 对角线

引用中提到了使用NumPy库中的预定义函数numpy.diag(a)来找到矩阵的对角线元素。这个函数会自动将对角线元素存储在一个数组中。下面是一个示例代码，展示了如何使用numpy.diag()函数来找到一个矩阵的对角线元素：

import numpy as np

# 创建一个3x3的矩阵
a = np.arange(9).reshape((3,3))
print('矩阵 a ：', a)

# 打印矩阵的对角线元素
print('矩阵 a 的对角线元素：', np.diag(a))

这段代码中，我们首先创建了一个3x3的矩阵a，然后使用np.diag()函数找到了矩阵a的对角线元素并进行打印输出。