Pandas data frame to dictionary of lists

2020-04-07 02:33发布

问题:

How to use python or pandas (preferably) to convert a pandas data_frame to dictionary of lists for input into highcharts.

The closest I got to was (see tmp notebook here: https://b9ad8d-iad-002.tmpnb.org/user-fVD4OAy9K8wR/notebooks/pd_to_json.ipynb)

df.T.to_json('bar.json', orient='index') But this is a 'dict of dicts' instead of 'dict of lists'

My input:

import pandas
import numpy as np

df = pandas.DataFrame({
    "date": ['2014-10-1', '2014-10-2', '2014-10-3', '2014-10-4', '2014-10-5'],
    "time" : [1,2,3,4,5],
    "temp" : np.random.random_integers(0,10,5),
    "foo" : np.random.random_integers(0,10,5)
})

df2 = df.set_index(['date'])
df2

Desired Output: I am using this output in Highcharts, which requires it to be a 'dictionary of lists' like so:

{
"date": [
    "2014-10-1",
    "2014-10-2",
    "2014-10-3",
    "2014-10-4",
    "2014-10-5"
],
"foo": [
    7,
    2,
    5,
    5,
    6
],
"temp": [
    8,
    6,
    10,
    10,
    3
],
"time": [
    1,
    2,
    3,
    4,
    5
]

}

Any help is much appreciated!

回答1:

In [199]: df2.reset_index().to_dict(orient='list')
Out[199]: 
{'date': ['2014-10-1', '2014-10-2', '2014-10-3', '2014-10-4', '2014-10-5'],
 'foo': [8, 1, 8, 8, 1],
 'temp': [10, 10, 8, 3, 10],
 'time': [1, 2, 3, 4, 5]}


回答2:

to create a list of the dictionaries per line

post_data_list = []
for i in df2.index:
  data_dict = {}
  for column in df2.columns:
    data_dict[column] = df2[column][i]
  post_data_list.append(data_dict)