Viết hàm số UnitMatrix(n) với n là số tự nhiên cho trước, hàm trả lại giá trị là ma trận bậc n như Hình 17.1.

def UnitMatrix(n):

    matrix = []

    for i in range(n):

        row = []

        for j in range(n):

            if i == j:

                row.append(1)

            else:

                row.append(0)

        matrix.append(row)

    return matrix