numpy.diag_indices_from

numpy.diag_indices_from(arr) [source]

Return the indices to access the main diagonal of an n-dimensional array.

See diag_indices for full details.

Parameters

arrarray, at least 2-D

See also

diag_indices

Notes

New in version 1.4.0.