Numpy：如何在Numpy中创建单位矩阵

在本文中，我们将介绍如何在Numpy中创建单位矩阵。单位矩阵，也被称为恒等矩阵，是对角线上元素为1，其余元素为0的方阵。在矩阵运算中，单位矩阵通常作为乘法单位元存在。

阅读更多：Numpy 教程

创建单位矩阵

在Numpy中，可以使用numpy.identity()函数来创建单位矩阵。该函数接受一个整数参数n，返回一个n \times n的单位矩阵。下面是一个简单的示例，演示如何使用numpy.identity()函数创建一个3 \times 3的单位矩阵：

import numpy as np

I = np.identity(3)
print(I)

该代码将输出：

array([[1., 0., 0.],
       [0., 1., 0.],
       [0., 0., 1.]])

可以看到，生成的矩阵满足对角线上元素为1，其余元素为0的条件。

此外，还可以使用numpy.eye()函数来创建单位矩阵。numpy.eye()函数与numpy.identity()函数类似，但可以接受两个整数参数m和n，返回一个m \times n的单位矩阵。下面是一个示例，演示如何使用numpy.eye()函数创建一个2 \times 3的单位矩阵：

import numpy as np

I = np.eye(2, 3)
print(I)

该代码将输出：

array([[1., 0., 0.],
       [0., 1., 0.]])

可以看到，生成的矩阵满足对角线上元素为1，其余元素为0的条件。注意，由于m，因此生成的矩阵只有前两列是单位矩阵，而第三列的元素都为0。

总结

在本文中，我们介绍了如何在Numpy中创建单位矩阵。通过使用numpy.identity()函数或numpy.eye()函数，可以轻松地生成对角线上元素为1，其余元素为0的方阵。这些函数在矩阵计算和线性代数中经常被使用，对于学习和实践这些领域的人来说是非常有用的工具。