numpy.ma.identity()

numpy.ma.identity

numpy.ma.identity(n, dtype=None) =

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 diago