def f(x):
	return [[1 if n == z else 0 for z in range(x)] for n in range(x)]
 
print(f(5))
print(f(20))