'Replacing diagonal elements with another array in Python

I have two matrices, A and P. I would like to replace the diagonal elements of A with the elements of P. The desired output is attached.

A=np.array([[0, 1, 1, 0],
       [1, 0, 0, 1],
       [1, 0, 0, 1],
       [0, 1, 1, 0]])

P=np.array([[3, 4],
       [5, 6]])

Desired output:

array([[3, 1, 1, 0],
       [1, 4, 0, 1],
       [1, 0, 5, 1],
       [0, 1, 1, 6]])

python numpy

Solution 1:^[1]

Use fill_diagonal with the flattened P as values for the diagonal:

np.fill_diagonal(A, P.ravel())

NB. the operation is in place

output:

array([[3, 1, 1, 0],
       [1, 4, 0, 1],
       [1, 0, 5, 1],
       [0, 1, 1, 6]])

This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.

Solution	Source
Solution 1	mozway