使用 Python 创建单位矩阵

在线性代数中，n 维单位矩阵是一个 n × n 方阵，主对角线上为 1，其他处为 0。本文将解释如何使用 Python 编程语言的 NumPy 库创建单位矩阵。

使用 Python 创建单位矩阵

NumPy 是一个 Python 编程语言库，用于创建大型多维数组和矩阵。使用 pip3 install numpy 命令安装 NumPy 库以创建单位矩阵。

这个库的 identity() 函数将一个数字作为参数，并返回一个具有该行数和列数的标识数组。你可以选择指定输出的数据类型。它默认使用 float。

在下面的示例中创建了一个具有 integer 数据类型的 4x4 单位矩阵。

import numpy as np

matrix_int = np.identity(4, dtype = int)
print(matrix_int)

如果未指定 dtype 参数，则使用 float 数据类型创建矩阵。

import numpy as np

matrix_float = np.identity(4)
print(matrix_float)