遍历多个 pandas Dataframe 对象

Question

我有一个预测功能，我已将其应用于一个数据框，它工作正常。

output =forecast(rawdf)

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

我的目标是将 rawdf 分解为与 depot 对应的多个数据帧，并将预测功能分别应用于所有数据帧。

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

尝试相同的逻辑，一一传递 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)]