scipy.sparse.dia_array.diagonal#
- dia_array.diagonal(k=0)[source]#
Returns the kth diagonal of the array/matrix.
- Parameters:
- kint, optional
Which diagonal to get, corresponding to elements a[i, i+k]. Default: 0 (the main diagonal).
New in version 1.0.
See also
numpy.diagonal
Equivalent numpy function.
Examples
>>> from scipy.sparse import csr_array >>> A = csr_array([[1, 2, 0], [0, 0, 3], [4, 0, 5]]) >>> A.diagonal() array([1, 0, 5]) >>> A.diagonal(k=1) array([2, 3])