numpy.matlib.identity — NumPy v1.11 Manual (original) (raw)
numpy.matlib.identity(n, dtype=None)[source]¶
Returns the square identity matrix of given size.
Parameters: | n : int Size of the returned identity matrix. dtype : data-type, optional Data-type of the output. Defaults to float. |
---|---|
Returns: | out : matrix n x n matrix with its main diagonal set to one, and all other elements zero. |
See also
Equivalent array function.
matlib.eye
More general matrix identity function.
Examples
import numpy.matlib np.matlib.identity(3, dtype=int) matrix([[1, 0, 0], [0, 1, 0], [0, 0, 1]])