Python-Utility-Scripts/matrix.py at main · AcoAlexDev/Python-Utility-Scripts · GitHub

34 lines (26 loc) · 735 Bytes

import numpy as np

def calculateMatrixMultiplication(A:np.array, B:np.array) -> np.array:

rowsA = A.shape[0]

colsA = A.shape[1]

rowsB = B.shape[0]

colsB = B.shape[1]

C = np.zeros((rowsA, colsB), dtype=int)

for r in range(rowsA):

for c in range(colsB):

result = 0

for k in range(colsA):

result += A[r][k] * B[k][c]

C[r][c] = result

print(result)

return C

def main() -> None:

A = np.array([[3, -1, -2],

[2, 4, -2]

])

B = np.array([[2, -2],

[3, 1],

[-2, 4]

])

calculateMatrixMultiplication(A, B)

print("Done")

main()

Expand file tree