challenge-011/lubos-kolouch/python/ch-2.py


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

#!/usr/bin/env python
# -*- coding: utf-8 -*-

size = 4  # change this to the desired size
identity_matrix = [[0 for x in range(size)] for y in range(size)]

for i in range(size):
    identity_matrix[i][i] = 1

# print the identity matrix
for row in identity_matrix:
    print(row)