-
Notifications
You must be signed in to change notification settings - Fork 0
/
MatrixAddition.py
84 lines (78 loc) · 3.02 KB
/
MatrixAddition.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
74
75
76
77
78
79
80
M1Flag = True
M2Flag = True
iMat1, iMat2, iMat3, iMat4 = [], [] ,[] , []
while True:
iChoice = int(input("""Enter 1: Enter value in Matrix 1
2: Enter value in Matrix 2
3: Addition
4: Substraction
5: Display Matrix
6: Exit
=>"""))
if iChoice == 1:
row1 = int(input("Input Number of Rows:"))
col1 = int(input("Input Number of Cols:"))
if M2Flag == False:
if row1 != row2 or col1 != col2:
print("Warning !!! Dimension of Matrix 2 is " + str(row2)+ "x" + str(col2))
iMat1 = [[int(input('index['+str(i)+"]["+str(j)+"]:")) for j in range(col1)]for i in range(row1)]
M1Flag = False
elif iChoice == 2:
row2 = int(input("Input Number of Rows:"))
col2 = int(input("Input Number of Cols:"))
if M1Flag == False:
if row1 != row2 or col1 != col2:
print("Warning !!! Dimension of Matrix 1 is " + str(row1)+ "x" + str(col1))
iMat2 = [[int(input('index['+str(i)+"]["+str(j)+"]:")) for j in range(col2)]for i in range(row2)]
M2Flag = False
elif iChoice == 3:
if row1 == row2 and col1 == col2:
iMat3 = [[iMat1[i][j] + iMat2[i][j] for j in range(col2)]for i in range(row1)]
else:
print("Dimensions of Matrices are not same!!!")
print("Matrix 1: " + str(row1)+ "x" + str(col1))
print("Matrix 2: " + str(row2)+ "x" + str(col2))
elif iChoice == 4:
if row1 == row2 and col1 == col2:
iMat4 = [[iMat1[i][j] - iMat2[i][j] for j in range(col2)]for i in range(row1)]
else:
print("Dimensions of Matrices are not same!!!")
print("Matrix 1: " + str(row1)+ "x" + str(col1))
print("Matrix 2: " + str(row2)+ "x" + str(col2))
elif iChoice == 5:
dChoice = int(input("""Which Matrix do you want to see,
Enter 1, for Matrix 1
2, for Matrix 2
3, for Addition Matrix
4, for Substraction Matrix
=>"""))
if dChoice == 1 and len(iMat1) != 0:
print("Matrix 1:")
for i in range(row1):
for j in range(col1):
print(iMat1[i][j], end = " ")
print()
elif dChoice == 2 and len(iMat2) != 0:
print("Matrix 2:")
for i in range(row2):
for j in range(col2):
print(iMat2[i][j], end = " ")
print()
elif dChoice == 3 and len(iMat3) != 0:
print("Addition Matrix:")
for i in range(row1):
for j in range(col1):
print(iMat3[i][j], end = " ")
print()
elif dChoice == 4 and len(iMat4) != 0:
print("Substraction Matrix:")
for i in range(row2):
for j in range(col2):
print(iMat4[i][j], end = " ")
print()
else:
print('Wrong Choice, Maybe the Matrix is not initialize.')
elif iChoice == 6:
break
else:
print("Wrong Choice")