python - pandas 将 dataframe id 列转换为字典并将相应的列列出

Question

我想将我的数据框转换为一个字典，其中的键是一个 id 列。我想将具有相应 id 的所有相应的第二列值放入列表中。

示例数据:

s1 = pd.Series(['A','A','B'],name = 's1')
s2 = pd.Series(['1','2','2'],name = 's2')

df = pd.concat([s1,s2],axis = 1)

s1 s2
A  1
A  2
B  2

我想要的输出

我如何获得带有映射的字典:

d = {'A':[1,2],'B':[2]}

Answer 1

还有两个选项

df.groupby('s1').s2.apply(list).to_dict()
Out[44]: {'A': ['1', '2'], 'B': ['2']}

s=df.groupby('s1').s2.apply(list)
dict(zip(s.index,s))
Out[48]: {'A': ['1', '2'], 'B': ['2']}