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

• Z时代
• 2024-01-10
• 分类：综合

在这个程序中，我们将打印一个大小为 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]]

以上是 如何使用 Numpy 创建单位矩阵？ 的全部内容， 来源链接： utcz.com/z/361086.html