Pandas dataframe 到列表字典

Question

Having a dataframe of the form:

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

I want to convert it to a dictionary of lists of the form

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

I am looking at the documentation of the .to_dict method, but none of the options seem to do what I need.

Answer 1

Usual to_dict bases on index elements as key values, so transpose your dataframe and apply list by row followed by dictionary conversion

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

Out:

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

Answer 2

Use DataFrame.to_dict with list parameter:

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