Indentity Matrix in Python

indMatrix.py

m, n = 3 , 3
# Create matrix
# B = [ [ 0 for i in range(m)] for j in range(n) ]
A = [  [  1 if i==j else 0  for i in range(m)] for j in range(n)]
# Output: >>> [[1, 0, 0], [0, 1, 0], [0, 0, 1]]
print('\n',A[0],'\n',A[1],'\n',A[2],'\n')