VPF::Аппроксимация функции - Форум программистов

rs5630

Дата 19.1.2015, 16:54 (ссылка)

(нет голосов)

Загрузка ...

Новичок

Профиль
Группа: Участник
Сообщений: 4
Регистрация: 20.12.2014

Репутация: нет
Всего: нет

Дана функция y = x^2 + случайное значение от 0 до 0.3. Нужно построить среднюю кривую и среднеквадратичное отклонение. Желательно с помощью numpy. Нашел как построить среднее и СКО, но нужно, чтобы среднее было кривой как на графике ниже, а не прямой как у меня. Подскажите как изменить программу.

График: ...

Код


import numpy as np
import matplotlib.pyplot as plt
from math import *
x = np.linspace(0,3,20)
y = np.zeros(20)
a = np.random.normal(0,0.3,size=2E1)
for i in range(20):
    y[i] = x[i]**2 + a[i]
A = np.vstack([x, np.ones(len(x))]).T
m, c = np.linalg.lstsq(A, y)[0]
plt.plot(x, y, 'bo-', lw = 1, label=u'Function')
plt.plot(x, m*x + c, 'r', label='Average')
std = np.zeros(20)
for i in range(20):
    std[i] = (sqrt((y[i] - (m*x[i] + c))**2))
plt.errorbar(x, y, std, label=u'Std')
plt.legend(loc='best')
plt.title(u'Plots')
plt.xlabel('X')
plt.ylabel('Y')
plt.grid()
plt.show()

Это сообщение отредактировал(а) rs5630 - 20.1.2015, 00:32

1 Пользователей читают эту тему (1 Гостей и 0 Скрытых Пользователей)
0 Пользователей:
« Предыдущая тема \| Python: Общие вопросы \| Следующая тема »