微客导航 » 文章资讯 » 如何使用 Numpy 创建单位矩阵？

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

2023-06-16 11:53:05 525

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

算法

Step 1: Import numpy.
Step 2: Take dimensions as input from the user.
Step 3: Print the identity matrix using numpy.identity() function.

import numpy as np

dimension = int(input("输入恒等矩阵的维数： "))
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]]