Python创建一个对角线为1的矩阵

在Python中，我们可以使用numpy库来创建一个数组或者矩阵，并且可以指定其形状和值。

下面是一个使用numpy库创建一个对角线为1的矩阵的示例代码：

import numpy as np

# 创建一个3x3的对角线为1的矩阵
matrix = np.eye(3, dtype=int)

print(matrix)

这段代码首先导入了numpy库，然后使用np.eye()函数创建了一个3x3的对角线为1的矩阵，并将其赋值给变量matrix。最后，使用print()函数输出了该矩阵的值。

下面是该代码的输出结果：

[[1 0 0]
 [0 1 0]
 [0 0 1]]

可以看到，该矩阵的对角线上的元素都是1，其余元素都是0，符合要求。

除了np.eye()函数外，我们还可以使用np.diag()函数创建一个对角线为1的矩阵，如下：

import numpy as np

# 创建一个3x3的对角线为1的矩阵
matrix = np.diag([1, 1, 1])

print(matrix)

此时，输出结果与前面的方法相同。

综上所述，使用numpy库创建一个对角线为1的矩阵非常简单，只需要调用相应的函数即可。