Pandas dataframe 到列表字典

Question

具有以下形式的 dataframe：

element1   | element2  |  element3
cat           bird          plane
dog           poodle        cloud
pig           bike          sky

我想将其转换为表单列表的字典

{ 'element1' : ['cat', 'dog', 'pig'], 
  'element2' : ['bird', 'poodle', 'bike'],
  'element3' : ['plane', 'cloud', 'sky']  }

我正在查看.to_dict方法的文档，但似乎没有一个选项可以满足我的需要。

Answer 1

通常 to_dict 基于索引元素作为键值，因此转置您的 dataframe 并逐行应用列表，然后进行字典转换

df.T.apply(list,1).to_dict()

出去：

{'element1': ['cat', 'dog', 'pig'],
 'element2': ['bird', 'poodle', 'bike'],
 'element3': ['plane', 'cloud', 'sky']}

Answer 2

使用带有list参数的DataFrame.to_dict ：

d = df.to_dict('list')
print (d)
{'element1': ['cat', 'dog', 'pig'],
 'element2': ['bird', 'poodle', 'bike'], 
 'element3': ['plane', 'cloud', 'sky']}