Python matrice

Prikazati matricu 10 x 10. Na glavnu dijagonalu postaviti jedinice, a ostali članovi biće nule.

mat = [0] * 10
for i in range(10):
    mat[i] = [0] * 10


for i in range(10):
    for j in range(10):
        if i == j:
            mat[i][j] = 1
        else:
            mat[i][j] = 0


for i in range(10):
    red = ""
    for j in range(10):
        red += str(mat[i][j]) + " "
    print(red)