基于列中的重复值拆分数据帧

Question

以下是我的起始数据帧：

StartDF = pd.DataFrame({'A': {0: 1, 1: 1, 2: 2, 3: 4, 4: 5, 5: 5, 6: 5, 7: 5}, 'B': {0: 2, 1: 2, 2: 4, 3: 2, 4: 2, 5: 4, 6: 4, 7: 5}, 'C': {0: 10, 1: 1000, 2: 250, 3: 100, 4: 550, 5: 100, 6: 3000, 7: 250}})

我需要根据列a和列B中的重复值创建单个数据帧的列表，因此它应该如下所示：

^{pr2}$

我看到了很多解释如何删除重复项的答案，但是我需要保留重复值，因为无论a列和B列中的重复项如何，C列中的信息通常在行之间是不同的。所有行数据都需要保留在新的数据帧中。在

另外要注意的是，起始数据帧（StartDF）的长度会发生变化，所以每次运行这个时，创建的单个数据帧的数量都是可变的。最后，我需要将新创建的数据帧打印到它们自己的csv文件中（我知道如何完成这一部分）。只需要知道如何以优雅的方式从原始数据帧中取出数据。在

Answer 1

您可以使用^{}，迭代每个组并使用列表理解构建列表。在

df_list = [g for _, g in df.groupby(['A', 'B'])]

^{pr2}$