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)>