scipy.sparse.

eye_array#

scipy.sparse.eye_array(m, n=None, *, k=0, dtype=<class 'float'>, format=None)[source]#

Identity matrix in sparse array format

Return a sparse array with ones on diagonal. Specifically a sparse array (m x n) where the kth diagonal is all ones and everything else is zeros.

Parameters:
mint or tuple of ints

Number of rows requested.

nint, optional

Number of columns. Default: m.

kint, optional

Diagonal to place ones on. Default: 0 (main diagonal).

dtypedtype, optional

Data type of the array

formatstr, optional (default: “dia”)

Sparse format of the result, e.g., format=”csr”, etc.

Examples

>>> import numpy as np
>>> import scipy as sp
>>> sp.sparse.eye_array(3).toarray()
array([[ 1.,  0.,  0.],
       [ 0.,  1.,  0.],
       [ 0.,  0.,  1.]])
>>> sp.sparse.eye_array(3, dtype=np.int8)
<DIAgonal sparse array of dtype 'int8'
    with 3 stored elements (1 diagonals) and shape (3, 3)>