`numpy:diag_indices_from`

Signature

def diag_indices_from ( arr )

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

See diag_indices for full details.

import numpy as np

✓

Create a 4 by 4 array.

a = np.arange(16).reshape(4, 4)
a

✓

Get the indices of the diagonal elements.

di = np.diag_indices_from(a)
di

✓

a[di]

✓

This is simply syntactic sugar for diag_indices.

np.diag_indices(a.shape[0])

✓