exl两列数据，以第一列内容为分组把第二列数据读取到多维数组中，请教该如何实现！

Question

问题遇到的现象和发生背景

exl内容

A列 B列
C1 A0001
C1 A0002
C1 A0003
C1 A0004
C2 A0001
C2 B0001
C3 A0001
C3 A0002
C3 A0003
C3 A0004
C4 A0002
C4 B0001
C4 B0002
C5 D0001
C5 D0002
想以第一列的内容为分组，把第二列数据读取到多维数组中。

我想要达到的结果

[['A0001', 'A0002', 'A0003', 'A0004'], ['A0001', 'B0001'], ['A0004', 'A0003', 'A0002', 'A0001'], ['A0002', 'B0001', 'B0002'], ['D0001', 'D0002']]

Answer 1

import pandas as pd
data = pd.read_excel('data.xlsx')
result=list(data.groupby('A').apply(lambda x: list(x['B'])).values)

如有帮助，点一下下采纳