Numpy.eye()介绍

简介

numpy.eye()函数返回一个给定大小的单位矩阵。单位矩阵是一个正方形的矩阵，主对角线元素为1，其余元素均为0。

语法

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

参数解释：

• N：生成的矩阵的行数（int类型）
• M：（可选参数）生成的矩阵的列数，默认与行数相等。（int类型）
• k：（可选参数）对角线的偏移量，默认为0。0表示主对角线，正数为在主对角线上方，负数为在主对角线下方。（int类型）
• dtype：（可选参数）返回矩阵中元素的数据类型，默认为float。（数据类型类）
• order：（可选参数）返回数组中数据存放的顺序，'C'表示行优先，'F'表示列优先。默认是'F'。（{'C', 'F'}, optional）

返回值

返回一个NxN的矩阵，当指定M时返回一个NxM的矩阵。

示例代码

import numpy as np

# 创建一个3x3的单位矩阵
A = np.eye(3)
print(A)

# 创建一个2x5的单位矩阵
B = np.eye(2, 5)
print(B)

# 创建一个3x3，主对角线向右偏移1个单位的单位矩阵
C = np.eye(3, k=1)
print(C)

# 创建一个4x4，主对角线向左偏移2个单位，数据类型为int的单位矩阵
D = np.eye(4, k=-2, dtype=int)
print(D)

输出：

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

总结

numpy.eye()函数用于创建单位矩阵，方便地通过参数调整偏移量、数据类型和数据存放顺序等参数来满足不同情况下的需求。