numpy.identity#

numpy.identity(n,dtype=None,*,like=None)[source]#

Return the identity array.

The identity array is a square array with ones onthe main diagonal.

Parameters:

nint

Number of rows (and columns) inn xn output.

dtypedata-type, optional

Data-type of the output. Defaults tofloat.

likearray_like, optional

Reference object to allow the creation of arrays which are notNumPy arrays. If an array-like passed in aslike supportsthe__array_function__ protocol, the result will be definedby it. In this case, it ensures the creation of an array objectcompatible with that passed in via this argument.

New in version 1.20.0.

Returns:

outndarray

n xn array with its main diagonal set to one,and all other elements 0.

Examples

Try it in your browser!

>>>importnumpyasnp>>>np.identity(3)array([[1.,  0.,  0.],       [0.,  1.,  0.],       [0.,  0.,  1.]])

Go BackOpen In Tab