PY
py
1
2
3
4
5
6
7
8
9
10
11
12
import numpy as np
n, p = [int(x) for x in input().split()]
X = []
for i in range(n):
X.append([float(x) for x in input().split()])
y = [float(x) for x in input().split()]
res = ((np.linalg.matrix_power(((np.transpose(X)) @ X),(-1))) @ (np.transpose(X)) @ y).round(2)
print(res)
Enter to Rename, Shift+Enter to Preview
OUTPUT
Run