How can code be optimized with the numpy library?

Question

Python

The vector x=1:3 is given. Create a matrix of the 3rd order, on the diagonals of which the elements of the vector would be. If the diagonal is shorter than size(x), then the filling starts at the 1st element of the vector x.

def replace(diagonal, matrix):
        matrix[0][0] = diagonal[0]
        try:
            matrix[1][1] = diagonal[1]
        except IndexError:
            pass
        try:
            matrix[2][2] = diagonal[2]
        except IndexError:
            pass
 
        return matrix

How else can you solve this problem without using loops? Maybe there is an easier way to solve it?

Answer 1

It seems that with this command you can
numpy.diag (v), where v is your vector.
Details can be found here
https://numpy.org/doc/stable/reference/generated/n...

Answer 2

Filling with the last three elements from the diagonal vector:

matrix[np.diag_indices_from(matrix)] = diagonal[-min(matrix.shape):]