n = 4 a = [[0]*n for x in range(n)] for i in range(n): for j in range(n): if i==j: a[i][j] = i+1