Как создать функцию, которая получает значение из списка и помещает в метод и собирает результаты?

Question

Я строю регрессионную модель и хочу построить прогнозы на графике, поэтому сначала мне нужно собрать эти прогнозы.

Я могу получить их, используя lin_reg_2.predict(poly_reg.fit_transform([[len(cases)+i]])), когда i должно быть 1, 2, 3, ... et c.

Я создал список с именем i, диапазон которого составляет 1-30.

И теперь мне нужно создать функцию, которая будет получать каждый элемент из i, вставлять их в метод, получить результаты и сохранить их в список.

Определенно не знаю, как создать эту функцию. Кажется, это легко, но я даже не знаю, где я могу узнать о создании функций правильным способом.

Answer 1

Ну, это было проще, чем я думал:

predictions =np.array([[]])
n = np.array([i for i in range(1, 30)]).astype(float)
b = len(cases) + n
b  = b.reshape(b.shape[0], -1)
for n in lin_reg_2.predict(poly_reg.fit_transform(b)):
    predictions = lin_reg_2.predict(poly_reg.fit_transform(b))