如何使用Numpy创建单位矩阵？

在此程序中，我们将打印一个大小为nxn的单位矩阵，其中n将从用户输入。我们将使用numpy库中的identity()函数，该函数将尺寸和元素数据类型作为参数。

算法

步骤1：导入numpy。
步骤2：从用户输入获取维度。
步骤3：使用numpy.identity()函数打印单位矩阵。

示例代码

import numpy as np

dimension = int(input("Enter the dimension of identitiy matrix: "))
identity_matrix = np.identity(dimension, dtype="int")
print(identity_matrix)

输出

输入单位矩阵的维度：5
[[1 0 0 0 0]
 [0 1 0 0 0]
 [0 0 1 0 0]
 [0 0 0 1 0]
 [0 0 0 0 1]]