python - 将pandas groupby对象转换为数据框列表

Question

假设我有以下数据框，并希望按ys分组:

   xs  ys
0   0   0
1   1   0
2   2   1
3   3   1

我可以通过运行来做到这一点

grouped = df.groupby('ys')

我可以很好地遍历这个新的groupby对象，但是我想要一个由 group在以下循环中访问的数据帧的列表:

for name, group in grouped:
    do_something(group)

这可能吗？

Answer 1

当然，只需遍历各个组!

>>> import pandas as pd, numpy as np
>>> df = pd.DataFrame(dict(xs=list(range(4)), ys=[0,0,1,1]))
>>> df
   xs  ys
0   0   0
1   1   0
2   2   1
3   3   1
>>> grouped = df.groupby('ys')
>>> dataframes = [group for _, group in grouped]
>>> dataframes
[   xs  ys
0   0   0
1   1   0,    xs  ys
2   2   1
3   3   1]
>>>