changing the values of the diagonal of a matrix in numpy

Question

how can I change the values of the diagonal of a matrix in numpy?

I checked Numpy modify ndarray diagonal, but the function there is not implemented in numpy v 1.3.0

Answer 1

You can do the following.

Assuming your matrix is 4 * 4 matrix.

indices_diagonal = np.diag_indices(4)

yourarray[indices_diagonal] = Val