关键词

详解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。

本文链接:http://task.lmcjl.com/news/4428.html

展开阅读全文