python - Numpy modify ndarray diagonal

Question

is there any way in numpy to get a reference to the array diagonal? I want my array diagonal to be divided by a certain factor Thanks

Answer 1

If X is your array and c is the factor,

X[np.diag_indices_from(X)] /= c

See diag_indices_from in the Numpy manual.