numpy.identity()

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

Return the identity array.

The identity array is a square array with ones on the main diagonal.

Parameters:

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.

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.

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

Links:

doc_NumPy

2017-01-10 18:14:23

Comments