numpy.identity — NumPy v1.13 Manual (original) (raw)
numpy. identity(n, dtype=None)[source]¶
Return the identity array.
The identity array is a square array with ones on the main diagonal.
| Parameters: | n : int Number of rows (and columns) in n x n output. dtype : data-type, optional Data-type of the output. Defaults to float. |
|---|---|
| Returns: | out : ndarray n x n array with its main diagonal set to one, and all other elements 0. |
Examples
np.identity(3) array([[ 1., 0., 0.], [ 0., 1., 0.], [ 0., 0., 1.]])