n * n matrix diagonal of the array.

'''
example 
Enter your value:5
[[1. 0. 0. 0. 0.]
 [0. 2. 0. 0. 0.]
 [0. 0. 3. 0. 0.]
 [0. 0. 0. 4. 0.]
 [0. 0. 0. 0. 5.]] '''
 
 import numpy as np

input_val = int(input('Enter your value:'))

x = np.zeros((input_val, input_val))

np.fill_diagonal(x, range(1,input_val+1))

print(x)