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()函数的用法:
import numpy as np
arr = np.eye(3)
print(arr)
输出结果为:
[[1. 0. 0.]
[0. 1. 0.]
[0. 0. 1.]]
可以看到,该代码创建了一个3*3的单位矩阵,主对角线上的元素都是1,其他元素都是0。
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。
本文链接:http://task.lmcjl.com/news/4428.html