diagonal — SciPy v1.15.2 Manual (original) (raw)

scipy.sparse.csc_array.

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

Added in version 1.0.

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