Как рассчитать градиент наиболее подходящей линии в python? У меня есть 2 массива x и y, которые я начертил, а затем сделал наилучшую линию с помощью полифита (нашел пример в Интернете). Сейчас я пытаюсь найти градиент моей наилучшей линии, но я не знаю, как это сделать. Я пробовал смотреть на подобные вопросы здесь, но ничего из того, что я пробовал, пока не сработало. Вот мой код:
x = np.array(a)
y = np.array(b)
plt.figure()
plt.plot(x, y)
plt.plot(np.unique(x), np.poly1d(np.polyfit(x, y, 1))(np.unique(x)), color = 'k')
plt.xlim((0,100))
plt.ylim((0,100))
plt.show()
Затем я попытался сделать это, как я видел по аналогичному вопросу:
m = ((np.mean(x) * np.mean(y)) − (np.mean(x * y))/((np.mean(x))**2 − np.mean(x**2))
print(m)
Но я просто получаю ошибки. Спасибо.
Наклон уже возвращается функцией полифита. Так:
slope, intercept = np.polyfit(x, y, 1)
#For a linear polynomial (so ,1), the formula for the line = slope*x+intercept (ax+b)
Применительно к вашему делу:
import numpy as np
import matplotlib.pyplot as plt
#Example data
x = [0,1,2,3,4]
y = [1,2,5,8,1]
x = np.array(x)
y = np.array(y)
#Fit line
slope, intercept = np.polyfit(x, y, 1)
#Plot
plt.figure()
plt.scatter(x, y)
plt.plot(np.unique(x), np.poly1d(np.polyfit(x, y, 1))(np.unique(x)), color = 'k')
plt.show()
print(slope)
Пожалуйста, обратитесь к https://numpy.org/doc/stable/reference/generated/numpy.polyfit.html