如何使用 Numpy 创建单位矩阵?
在这个程序中,我们将打印一个大小为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]]