如何按列的值对pandas数据帧的行进行分组？

Question

如何按列的值对pandas数据帧的行进行分组？

假设我们有一个名为df的数据框：

A   B  C
1   1a 1b
1   1c 1d
1   1e 1f
2   2a 2b
2   2c 2d
3   3a 3b
3   3c 3d

我想使用groupby创建以下内容：

1: {[1a, 1b],
    [1c, 1d],
    [1e, 1f]}

2: {[2a,2b],
    [2c, 2d]}


3: {[3a,3b],
    [3c. 3d]}

我确实知道.loc是一个选项。但对于我正在研究的超大型数据集来说，这是非常缓慢的。这就是为什么我认为把它放在列表字典中可能会更好。

感谢。

Answer 1

似乎你需要：

df = df.groupby('A')['B','C'].apply(lambda x: x.values.tolist()).to_dict()
print (df)
{1: [['1a', '1b'], ['1c', '1d'], ['1e', '1f']], 
 2: [['2a', '2b'], ['2c', '2d']], 
 3: [['3a', '3b'], ['3c', '3d']]}