如何使用python的pandas从数据帧创建字典字典

Question

我有以下直接来自SQL查询的数据帧：

Question            tagID   Answer  
Primary purpose     62      Other  
Primary purpose     226     Learn how to use  
Primary purpose     227     Technical Support  
Primary purpose     292     Purchase  
Language            246     English  
Language            247     French   
Language            248     German  
Device              102     Desktop  
Device              103     Tablet  
Device              104     Mobile  

我需要一本字典，比如：

^{pr2}$

我尝试了以下代码，但它列出了所有值和标签：

    SS_valueLabelsSQL = {}
    for q in df['Question']:
       SS_valueLabelsSQL[q] = {}
       labels = df['Answer'].tolist()
       values = df['tagID'].tolist()
       SS_valueLabelsSQL[q] = dict(zip(values,labels))

有人能提出更好的解决方案吗？在

Answer 1

您可以使用：

df.set_index('Question').groupby(level='Question').apply(lambda x: x.set_index('tagID').squeeze().to_dict()).to_dict()

获得：

^{pr2}$