遍历多个数据帧对象

Question

我有一个预测功能，我已经应用到一个数据帧，它工作得很好

output =forecast(rawdf)

问题是我可以提取多个数据帧，如果我想使用相同的函数来运行所有的数据帧。因为rawdf有一个名为deptname的列，它有多个唯一的值

我的目标是将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

用相同的逻辑逐个传递deoptlist值

instead of passing list in isin(), pass value by iterating list

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

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

结果将包含预测值

Dont't reassign same dataframe variable of result rawdf.loc[rawdf.DepotName.isin(Depot)]