将基于熊猫python中重复列的数据集分组到列表中

Question

我有一个从excel导入到熊猫的数据框，格式如下：

df = pd.read_excel('excel_file.xlsx')

data = pd.DataFrame(df, columns=['A', 'B', 'C', 'D', 'E'])


A    B   C    D   E
12  bob  32  abc 123
12  jan  34  fbc  23
14  jan  32  ac  133
12  cat  32  abc 123

我想将它们按B列分组，以便输出为：

list[0] = [[12  bob  32  abc 123]]
list[1] = [[12  jan  34  fbc  23][14  jan  32  ac  133]]
list[2] = [[12  cat  32  abc 123]]

我尝试使用重复函数没有成功

非常感谢！

Answer 1

您可以做：

lst = [d.values.tolist() for (k,d) in df.groupby('B', sort=False)]

# check
for i in range(len(lst)): print(lst[i])

输出：

[[12, 'bob', 32, 'abc', 123]]
[[12, 'jan', 34, 'fbc', 23], [14, 'jan', 32, 'ac', 133]]
[[12, 'cat', 32, 'abc', 123]]