将dataframe中的两列数据转换成字典dic

df 为dataframe对象,包含'word', 'count'两列数据

 

方法一

result_dic = df.groupby('word')['count'].apply(list).to_dict() # 对于同一个key对应多个value,则把同一key的value构成一个list

result_dic = df.groupby('word')['count'].apply(lambda x:str(x)).to_dict()  # 对于同一个key对应一个value,直接使用value

结果

形如{key:value}的字典

 

方法二

result_dic = df.set_index('word').T.to_dict()

结果

形如{key:{'count':value}}

 

 

注:

result_dic = df.groupby('word')['count'].apply(lambda x:str(x)).to_dict()  # 如果把str(x)换成x,结果完全不一样,目前还不是很理解。

你可能感兴趣的:(Python学习)