python怎么创建对角矩阵

在Python中，可以使用numpy库中的diag函数来创建对角矩阵。具体方法如下：

import numpy as np

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

print(diag_matrix)

输出结果为：

array([[1, 0, 0],
       [0, 2, 0],
       [0, 0, 3]])

其中，np.diag函数的参数为一个列表，列表中的元素为对角线上的元素。如果需要创建一个n阶的单位矩阵，可以使用np.eye(n)函数。