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 dtype : data-type, optional Data-type of the output. Defaults to |
---|---|
Returns: |
out : ndarray
|
>>> np.identity(3) array([[ 1., 0., 0.], [ 0., 1., 0.], [ 0., 0., 1.]])
© 2008–2017 NumPy Developers
Licensed under the NumPy License.
https://docs.scipy.org/doc/numpy-1.14.2/reference/generated/numpy.identity.html