NumPy参考 >例行程序 >Matrix library (numpy.matlib) > numpy.matlib.identity
numpy.matlib.
identity
(n,dtype = None )[源代码] ¶返回给定大小的平方单位矩阵。
返回的身份矩阵的大小。
输出的数据类型。默认为float
。
n x n矩阵,其主对角线设置为1,所有其他元素设置为零。
也可以看看
numpy.identity
等效数组函数。
matlib.eye
更通用的矩阵标识函数。
例子
>>> import numpy.matlib
>>> np.matlib.identity(3, dtype=int)
matrix([[1, 0, 0],
[0, 1, 0],
[0, 0, 1]])