NumPy参考 >例行程序 >Matrix library (numpy.matlib) > numpy.matlib.eye
numpy.matlib.
eye
(n,M = None,k = 0,dtype = <class'float'>,order ='C' )[源代码] ¶返回一个矩阵,在对角线上有一个,在其他地方为零。
输出中的行数。
输出中的列数,默认为n。
对角线的索引:0表示主对角线,正值表示上对角线,负值表示下对角线。
返回矩阵的数据类型。
输出应按行优先(C样式)还是列优先(Fortran样式)的顺序存储在内存中。
1.14.0版中的新功能。
一个n x M矩阵,其中除第k个对角线(其值等于1)外,所有元素均等于0 。
例子
>>> import numpy.matlib
>>> np.matlib.eye(3, k=1, dtype=float)
matrix([[0., 1., 0.],
[0., 0., 1.],
[0., 0., 0.]])