首页 > 解决方案 > 要 dict 的数据框,其中一列是键,另一列是值

问题描述

我有数据框

    ID   A   B   C
0   p    1   3   2
1   q    4   3   2
2   r    4   0   9  

我想创建一个字典,其中 ID 是键,B 是值,所以它将是:

d["q"] = 3 , d["r"] = 0

最好的方法是什么?

它与假设的重复不同,因为我想要每个键的单个值而不是列表

标签: pythonpandasdictionarydataframe

解决方案


像这样的东西:

In [27]: df
Out[27]: 
  ID  A  B  C
0  p  1  3  2
1  q  4  3  2
2  r  4  0  9

In [30]: df.set_index('ID',inplace=True)
In [31]: df
Out[31]: 
    A  B  C
ID         
p   1  3  2
q   4  3  2
r   4  0  9

In [33]: df.to_dict()['B']
Out[33]: {'p': 3, 'q': 3, 'r': 0}

推荐阅读