python - 从 Excel 到分割的 Python 结构

Question

我有以下 Excel 文件:

ID    Name     Budget
...   ...      ...
...   ...      ...

... some unfilled blank cells

ID    Name     Budget
...   ...      ...

... some unfilled blank cells

ID    Name     Budget
...   ...      ...

我想使用 Pandas(例如 ExcelFile)将这个 Excel 工作表读取为单独的结构(未填充单元格之前的每个表构成一个数据框/字典/...)。

我需要这样做，以便我可以处理同一结构中以及多个结构之间的数据(例如对每个结构中重复 ID 或名称的预算求和)

在保持合理的内存性能的同时，最简单的方法是什么？

Answer 1

下面是通过read_excel()读取所有数据并拆分的代码:

import pandas as pd

df = pd.read_excel("c:\\tmp\\book1.xlsx", "Sheet1")
mask = df["ID"] == "ID"
nmask = ~mask
s = mask.astype(int).cumsum()
dfs = [x.dropna() for _,x in df[nmask].groupby(s[nmask])]

for df in dfs:
    print df

dfs中的值都是对象。