Как я могу построить линейную регрессию с ошибками? - PullRequest
0 голосов
/ 10 февраля 2020

Я пытаюсь сделать линейную регрессию с барами ошибок, используя matplotlib. Я не знаю, как добавить строки ошибок.

Это мой код:

import numpy as np
import matplotlib.pyplot as plt

x = [6, 15, 24, 33, 41, 52, 59, 66, 73, 81]
y = [5,10,15,20,25,30,35,40,45,50]

coef = np.polyfit(x,y,1)
poly1d_fn = np.poly1d(coef) #to create a linear function with coefficients

plt.plot(x,y,'yo', x,  poly1d_fn(x), '-k')
plt.show()

1 Ответ

1 голос
/ 10 февраля 2020

Может быть как то так?

import matplotlib.pyplot as plt
import numpy as np

x = np.array([6, 15, 24, 33, 41, 52, 59, 66, 73, 81])
y = np.array([5, 10, 15, 20, 25, 30, 35, 40, 45, 50])

coef = np.polyfit(x, y, 1)
poly1d_fn = np.poly1d(coef)  # to create a linear function with coefficients

plt.plot(x, y, 'ro', x, poly1d_fn(x), '-b')
plt.errorbar(x, poly1d_fn(x), yerr=poly1d_fn(x) - y, fmt='.k')
plt.show()
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...