Создание столбцов данных в цикле for - PullRequest
0 голосов
/ 07 мая 2019

Мне нужна помощь по проблеме с пандами:

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

 date       number
2018-01-01     1
2018-02-01     2
2018-03-01     3
2018-04-01     3

И я вызываю эту функциюв течение цикла.Цикл for выполняет следующие действия:

for i in range(len(category_list)):
    print(category_list[i])
    serie = filter_users_pageNS(data, new_index, category_list[i])
    serie = pd.DataFrame(serie).reset_index()

Где category_list - это просто число итераций, которое должно быть выполнено с параметром, передаваемым в функцию filter_users_pageNS.

MyНамерение состоит в том, чтобы добавить столбец number, полученный на каждой итерации в фрейме данных, чтобы в конце цикла for я получил следующий фрейм данных:

date      number_iteration_1      number_iteration_2
2018-01-01       1                       4
2018-02-01       2                       1
2018-03-01       3                       5
2018-04-01       3                       2

Как видите, окончательный фрейм данныхдобавлен столбец number к кадру данных с именем number_iteration_x.

Есть идеи, как получить этот окончательный кадр данных?

Заранее большое спасибо.

1 Ответ

0 голосов
/ 07 мая 2019

IIUC, что-то вроде этого должно работать:

df=df.reset_index()
for i in range(len(category_list)):
    print(category_list[i])
    serie = filter_users_pageNS(data, new_index, category_list[i])
    serie = pd.DataFrame(serie).reset_index()
    df['number_iteration_'+str(i)]=serie 

Редактировать: скорректированный ответ на основе ваших новых данных

...