Compute matrix product of two arrays
import numpy as np n = int(input()) a = np.array([input().split() for _ in range(n)], int) b = np.array([input().split() for _ in range(n)], int) print(np.dot(a, b))