Question

How to fill matrix center diagonal with different values using numpy in Python

Answer 1

import numpy as np

matrix = np.arange(9).reshape((3,3))

np.fill_diagonal(matrix, [162, 371, 910])

print(matrix)



'''
run:

[[162   1   2]
 [  3 371   5]
 [  6   7 910]]

'''

 

70+ SQL courses for beginners and professionals