-
Notifications
You must be signed in to change notification settings - Fork 0
/
5_IDENTITY matrix Check.py
executable file
·73 lines (57 loc) · 1.7 KB
/
5_IDENTITY matrix Check.py
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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
# By Ashwath from forums
# Given a list of lists representing a n * n matrix as input,
# define a procedure that returns True if the input is an identity matrix
# and False otherwise.
# An IDENTITY matrix is a square matrix in which all the elements
# on the principal/main diagonal are 1 and all the elements outside
# the principal diagonal are 0.
# (A square matrix is a matrix in which the number of rows
# is equal to the number of columns)
def is_identity_matrix(matrix):
# check if square or not
if len(matrix) != len(matrix[0]):
return False
n = len(matrix)
for i in range(n):
for j in range(n):
if i == j and matrix[i][j] != 1:
return False
elif i != j and matrix[i][j] != 0:
return False
return True
# Test Cases:
matrix1 = [[1, 0, 0, 0],
[0, 1, 0, 0],
[0, 0, 1, 0],
[0, 0, 0, 1]]
print is_identity_matrix(matrix1)
# >>>True
matrix2 = [[1, 0, 0],
[0, 1, 0],
[0, 0, 0]]
print is_identity_matrix(matrix2)
# >>>False
matrix3 = [[2, 0, 0],
[0, 2, 0],
[0, 0, 2]]
print is_identity_matrix(matrix3)
# >>>False
matrix4 = [[1, 0, 0, 0],
[0, 1, 1, 0],
[0, 0, 0, 1]]
print is_identity_matrix(matrix4)
# >>>False
matrix5 = [[1, 0, 0, 0, 0, 0, 0, 0, 0]]
print is_identity_matrix(matrix5)
# >>>False
matrix6 = [[1, 0, 0, 0],
[0, 1, 0, 2],
[0, 0, 1, 0],
[0, 0, 0, 1]]
print is_identity_matrix(matrix6)
# >>>False
matrix7 = [[1, -1, 1],
[0, 1, 0],
[0, 0, 1]]
print is_identity_matrix(matrix7)
# >>>False