mirror of
https://github.com/TheAlgorithms/Python.git
synced 2024-11-24 13:31:07 +00:00
37 lines
850 B
Python
37 lines
850 B
Python
|
def add(matrix_a, matrix_b):
|
||
|
rows = len(matrix_a)
|
||
|
columns = len(matrix_a[0])
|
||
|
matrix_c = []
|
||
|
for i in range(rows):
|
||
|
list_1 = []
|
||
|
for j in range(columns):
|
||
|
val = matrix_a[i][j] + matrix_b[i][j]
|
||
|
list_1.append(val)
|
||
|
matrix_c.append(list_1)
|
||
|
return matrix_c
|
||
|
|
||
|
|
||
|
def multiply(matrix_a, matrix_b):
|
||
|
matrix_c = []
|
||
|
n = len(matrix_a)
|
||
|
for i in range(n):
|
||
|
list_1 = []
|
||
|
for j in range(n):
|
||
|
val = 0
|
||
|
for k in range(n):
|
||
|
val = val + matrix_a[i][k] * matrix_b[k][j]
|
||
|
list_1.append(val)
|
||
|
matrix_c.append(list_1)
|
||
|
return matrix_c
|
||
|
|
||
|
|
||
|
def main():
|
||
|
matrix_a = [[12, 10], [3, 9]]
|
||
|
matrix_b = [[3, 4], [7, 4]]
|
||
|
print(add(matrix_a, matrix_b))
|
||
|
print(multiply(matrix_a, matrix_b))
|
||
|
|
||
|
|
||
|
if __name__ == '__main__':
|
||
|
main()
|