numpy—np.eye、np.diag与np.tile

1.np.eye

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

描述

返回一个二维数组，对角线上为1，其他地方为0

参数

N : int
输出的行数

M : int, optional
输出中的列数。如果为None，默认为N

k : int, optional
对角线索引:0(默认)表示主对角线，正值表示上对角线，负值表示下对角线。

d : type,data-type, optional
返回数组的数据类型

order : {‘C’, ‘F’}, optional
在内存中，输出应该按行主要(c风格)还是列主要(fortran风格)顺序存储

返回

I : ndarray of shape (N,M)
一个数组，其中所有元素均等于零（第k个对角线除外，其值等于1）

实例

np.eye(2, dtype=int)

array([[1, 0],
       [0, 1]])


np.eye(3, k=1)

array([[0.,  1.,  0.],
       [0.,  0.,  1.