python - 通过对 pandas 中的列值进行分组来拆分 DataFrame

Question

我有一个数据框

S   C
3   2
3   2
3   2
3   2
3   2
3   2
1   4
1   4
1   4
1   4
1   4
1   4

我如何拆分数据帧，使一个数据帧在 S 和 C 中有 3 和 2，而另一个在 S 和 C 中有 1 和 4

Answer 1

使用 groupby 的推导式

1. 关键元素是 df.groupby，它提供您想要的分组。
2. 但是，您需要将其简化为一个列表，以便将其“拆分”为单独的数据框。
3. 您可以遍历 groupby 对象，该对象传递一个元组，其中第一个元素是组的名称(我们用 _ 屏蔽它)，第二个元素是单个数据框。<
4. 通过使用理解，我们可以遍历 groupby 并捕获每个元组的第二个元素...从而创建数据帧列表。

参见:
List Comprehensions
Grouping Stuff

---

list_of_df = [g for _, g in df.groupby(['NUMBER_OF_TRIPS', 'SERVICE_CLASS'])]

print(*list_of_df, sep='\n\n')

    NUMBER_OF_TRIPS  SERVICE_CLASS
6                 1              4
7                 1              4
8                 1              4
9                 1              4
10                1              4
11                1              4

   NUMBER_OF_TRIPS  SERVICE_CLASS
0                3              2
1                3              2
2                3              2
3                3              2
4                3              2
5                3              2