numpy.
diag_indices_from
Return the indices to access the main diagonal of an n-dimensional array.
See diag_indices for full details.
diag_indices
See also
Notes
New in version 1.4.0.