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。
本文链接：https://my.lmcjl.com/post/19854.html
展开阅读全文
4 评论