Задача C. Выделение тренда

Входной файл:Стандартный вход   Ограничение времени:1 сек
Выходной файл:Стандартный выход   Ограничение памяти:512 Мб

Условие

Пусть задан временной ряд {yt}nt=1. Требуется написать на языке Python функцию, вычисляющую коэффициенты функции тренда h(t)=mai(t).

def find_fit(x, y, f) # returns array of regression coefficients

Код решения должен содержать только определение и реализацию функции. Он не должен ничего выводить.

Формат входных данных

x, y — одномерные массивы типа np.ndarray. f — функция одного аргумента, возвращающая список значений функций gi, т.е. f(t)≡{gi(t)}mi=1.

Формат выходных данных

Функция должна возвращать единственный np.ndarray массив — коэффициенты ai.

Ограничения

1 < n < 103

1 < m ⩽ 5

Примеры тестов

Стандартный вход Стандартный выход
1
x = np.array([1, 2, 3, 4, 5])
y = np.array([2, 3, 5, 7, 11])
find_fit(x, y, lambda x: np.array((1, x)))
        
-1 2.2

0.106s 0.024s 15