Хорошо, так что идея состоит в том, чтобы построить оба столбца, один по оси X, а другой по оси Y, и попытаться создать линию, имитирующую его поведение.У Numpy есть функция для вычисления линии, поэтому
import numpy as np
import matplotlib.pyplot as plt
x = [4,2,1,5]
y = [2,4,6,3]
fit = np.polyfit(x,y,1)
fit_line = np.poly1d(fit)
plt.figure()
plt.plot(x,y,'rx')
plt.plot(x,fit_line(x),'--b')
plt.show()
И если мы считаем линию регрессии y = a*x + b
, вы можете получитькоэффициент a и b, так что
a = fit[0]
b = fit[1]
, который возвращает
a = -0.8000000000000005
b = 6.150000000000002
Просто используйте ваши x и y