NumPy 库的 eye 函数

NumPy 库的 eye 函数可以用来创建一个二维数组，其中主对角线为 1，其余元素为 0。eye 函数的语法如下：

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

其中，N 表示数组的行数，M 表示数组的列数（默认为 N），k 表示对角线的偏移量（默认为 0，表示主对角线），dtype 表示数组的数据类型，默认为 float，order 表示数组元素在内存中的存储顺序，可以是 'C'（按行存储）或 'F'（按列存储）。

例如，以下代码可以创建一个 3x3 的单位矩阵：

import numpy as np

arr = np.eye(3)
print(arr)

输出结果为：

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