python - 如何根据列行拆分数据框

Question

我有一个 excel 文件，dataframe 有 20 行。几行之后又是列名行，我想根据列名行划分数据框。

这里是例子:

x
0
1
2
3
4
x
23
34
5
6

预期的输出是:

df1
x
0
1
2
3
4
df2
x
23
34
5
6

Answer 1

考虑到您的列名是 col ，您可以首先将数据框分组为 cumsum在 col 上，其中值等于 x 的 df['col'].eq('x').cumsum() ，然后对于每个组通过使用 df.iloc[] 将该组第二行和列的值作为该组的第一个值来创建一个数据框，并将它们保存在字典中:

d={f'df{i}':pd.DataFrame(g.iloc[1:].values,columns=g.iloc[0].values) 
                   for i,g in df.groupby(df['col'].eq('x').cumsum())}

---

print(d['df1'])
 x
0  0
1  1
2  2
3  3
4  4

print(d['df2'])
    x
0  23
1  34
2   5
3   6