challenge-011/zapwai/python/ch-2.py


1
2
3
4
5
6
7
8
9
10
11
12

def id_mat(n):
    A = []
    for i in range(n):
        A.append([])
        for j in range(n):
            A[i].append(0)
    for i in range(n):
        A[i][i] = 1
    for i in range(n):
        print(A[i])
    
id_mat(3)