Question

How to fill only the diagonal of square list with numbers in Python

Answer 1

size = 7
lst = [[0] * size for i in range(size)]
for i in range(size):
    for j in range(size):
        if i == j:
            lst[i][j] = 1

for row in lst:
    print(' '.join([str(col) for col in row]))

'''
run:

1 0 0 0 0 0 0
0 1 0 0 0 0 0
0 0 1 0 0 0 0
0 0 0 1 0 0 0
0 0 0 0 1 0 0
0 0 0 0 0 1 0
0 0 0 0 0 0 1

'''

 

70+ SQL courses for beginners and professionals