Numpy 单位矩阵创建 eye()函数详解

eye()函数的参数设置如下

def eye(N, M=None, k=0, dtype=float, order='C', *, like=None):

N表示行数，M表示列数，k表示偏移，dtype为元素数据类型，order为存储顺序，like表示生成的为与like参数相同类型的数组（这个参数不常用，这里不做演示）

创建3行3列单位矩阵

import numpy as npdata1 = np.eye(3)
print(data1)# [[1. 0. 0.]
#  [0. 1. 0.]
#  [0. 0. 1.]]

一般来说单位矩阵为方阵，所以一般创建标准单位矩阵只需设置行或列数其一即可，但是这里设置行和列两个参数的原因是可以设置类单位矩阵（也就是即使矩阵为非方阵，对角线也默认从左上角开始）

import numpy as npdata1 = np.eye(N = 3,M = 4)
print(data1)# [[1. 0. 0. 0.]
#  [0. 1. 0. 0.]
#  [0. 0. 1. 0.]]

k参数的作用为空值对角线或类对角线的上移或下移，若k>0则上移，反之则下移，数值为移动的单位数

import numpy as npdata1 = np.eye(N = 3,M = 4,k=1)
data2 = np.eye(N = 5,M=4,k=-2)
print(data1,end='\n\n')
print(data2)# [[0. 1. 0. 0.]
#  [0. 0. 1. 0.]
#  [0. 0. 0. 1.]]
# 
# [[0. 0. 0. 0.]
#  [0. 0. 0. 0.]
#  [1. 0. 0. 0.]
#  [0. 1. 0. 0.]
#  [0. 0. 1. 0.]]

order参数为存储顺序，默认为‘C’，也就是按行顺序存储，也可以设置为‘F’，就是按列顺序存储

下面展示内存中的存储状态

import numpy as npdata1 = np.eye(N = 3,M = 4,k=1,order='F')
data2 = np.eye(N = 3,M = 4,k=1,order='C')print(data1.reshape(-1,order='F'))
print(data2.reshape(-1))
# [0. 0. 0. 1. 0. 0. 0. 1. 0. 0. 0. 1.]
# [0. 1. 0. 0. 0. 0. 1. 0. 0. 0. 0. 1.]