blob: 7402779a9973d4bdfbee8f0bc03fbfa527744208 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
|
#!/usr/bin/python3
# Challenge 011
#
# Challenge #2
# Write a script to create an Indentity Matrix for the given size. For example,
# if the size is 4, then create Identity Matrix 4x4. For more information about
# Indentity Matrix, please read the wiki page.
import sys
def id_matrix(size):
id = [[0 for y in range(size)] for x in range(size)]
for i in range(size):
id[i][i] = 1
return id
def print_matrix(m):
output = "["
sep = ""
for row in m:
output += sep+"["+", ".join([str(x) for x in row])+"]"
sep = ",\n "
output += "]"
print(output)
size = int(sys.argv[1])
id = id_matrix(size)
print_matrix(id)
|
