5行代码拟合直线

import numpy as np
#linalg:numpy线性代数库
from numpy import linalg
#x坐标(输入变量)
x = np.array([[1, float(i)] for i in input().split()])
#y坐标(输出坐标)
y = np.array([[float(i)] for i in input().split()])
#matmul:矩阵乘法
#inv:矩阵的逆
#.T:矩阵转置
print(np.matmul(np.matmul(linalg.inv(np.matmul(x.T, x)), x.T), y))