Итерация по нескольким объектам Dataframe панд

Question

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

output =forecast(rawdf)

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

Моя цель состоит в том, чтобы разбить rawdf на несколько фреймов данных, соответствующих депо, и применить функцию прогноза ко всем из них по отдельности.

DepotList= ['A', 'B']

for Depot in DepotList:
    i=1
    rawdf=rawdf.loc[rawdf.DepotName.isin([DepotList])]
    output[i] =forecast(rawdf)
    i = i+1

Я не пробовал, как будто ясобираюсь сделать это в первый раз.Я буду признателен за ваше время и усилия, чтобы помочь мне выполнить эту миссию.

Answer 1

Попробуйте одну и ту же логику со значением pass DepotList один за другим.

вместо проходного списка в isin () , передать значение с помощью итеративного списка

DepotList= ['A', 'B']
result = []
for Depot in DepotList:

    resultByDepotName =rawdf.loc[rawdf.DepotName.isin(Depot)]
    list = forecast(resultByDepotName)
    result.append(list)

результат будет содержать прогнозные значения

Не переназначать одну и ту же переменную кадра данных результата rawdf.loc [rawdf.DepotName.isin (Depot)]