通过（pandas）从组中的多个列创建字典

Question

我的数据框有＆＃39; id_one＆＃39;哪个可以有多个id_twos＆＃39;对于每个id。每个id_two还具有存储在其他列中的许多描述性特征。 这是一个示例数据集。

d = {'id_one' : pd.Series([123, 123, 123]),
     'id_two' : pd.Series([456, 567, 678]),
     'descriptor' : pd.Series(['blue','yellow', 'green'])}

df = pd.DataFrame(d)

我需要以每行{id_one＆＃39;一行的形式获取我的数据框，其中在＆＃39; col a＆＃39;我存储了id_one＆＃39;并且在＆＃39; col＆＃39;我存储了＆＃39; id_two＆＃39;的所有值。作为字典键和相应的描述符存储为字典值。

任何帮助将不胜感激，谢谢。

Answer 1

这是你要找的吗？

df.groupby('id_one').apply(lambda x: dict(zip(x['id_two'], x['descriptor']))).reset_index().rename(columns={"id_one":"col a", 0:"col b"})
#    col a                                          col b
# 0    123  {456: u'blue', 678: u'green', 567: u'yellow'}