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

Условие

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

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

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

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

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

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

Ограничения

1 < n < 10³

1 < m ⩽ 5

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

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