python pandas dataframe到字典

Question

我有两列数据框，打算将其转换为python字典-第一列将是键，第二列将是值。先感谢您。数据框：    id    value0    0     10.21    1      5.72    2      7.4

Answer 1

如果您想要一种简单的方法来保留重复项，则可以使用groupby：

>>> ptest = pd.DataFrame([['a',1],['a',2],['b',3]], columns=['id', 'value']) 

>>> ptest

  id  value

0  a      1

1  a      2

2  b      3

>>> {k: g["value"].tolist() for k,g in ptest.groupby("id")}

{'a': [1, 2], 'b': [3]}