Numpy矩阵运算 | np.eye()函数

简介

Numpy是Python中非常常用的科学计算库之一。其中，np.eye()函数可以生成一个指定大小的单位矩阵。

语法

np.eye(N, M=None, k=0, dtype=<class 'float'>, order='C')

• N：生成矩阵的行数（必填）。
• M：生成矩阵的列数（选填，默认为N）。
• k：矩阵对角线的偏移量（选填，默认为0，即主对角线）。
• dtype：矩阵的数据类型（选填，默认为float）。
• order：数据存储方式（选填，默认为行优先方式'C'）。

示例

import numpy as np

# 生成一个3行3列的单位矩阵
eye_3 = np.eye(3)
print(eye_3)

输出结果：

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

应用

np.eye()函数可以用于生成单位矩阵。在矩阵运算中，单位矩阵的特殊性质经常被用来做各种计算转换，比如矩阵求逆、矩阵转置等，因此是非常重要的一个概念。