详解Numpy eye()（返回对角线上为1的数组）函数的作用与使用方法

Numpy eye()函数是用来创建一个二维数组，其主对角线上的元素都是1，其他元素都是0。

该函数的语法如下：

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

其中，参数N表示二维数组的行数，M表示列数，默认为N；k表示主对角线偏移量，默认为0；dtype表示数据类型，可选值为float、int、complex等；order表示数组在内存中的存储顺序，可选值为'C'和'F'。

下面给出两个例子来说明Numpy eye()函数的用法：

例1：创建一个3*3的单位矩阵

import numpy as np

arr = np.eye(3)

print(arr)

输出结果为：

[[1. 0. 0.]
 [0. 1. 0.]
 [0. 0. 1.]]

可以看到，该代码创建了一个3*3的单位矩阵，主对角线上的元素都是1，其他元素都是0。

例2：创建一个3*4的右移2位的对角矩阵

import numpy as np

arr = np.eye(3, 4, k=2)

print(arr)

输出结果为：

[[0. 0. 1. 0.]
 [0. 0. 0. 1.]
 [0. 0. 0. 0.]]

可以看到，该代码创建了一个3*4的右移2位的对角矩阵，主对角线上的元素都是1，其他元素都是0。