Pandas数据框转换为列表字典

23

如何使用Python或Pandas(最好是Pandas)将Pandas DataFrame 转换为列表字典,以便输入到Highcharts?

我找到的最接近的方法是:

df.T.to_json('bar.json', orient='index')

但这是一个字典的字典,而不是列表的字典

我的输入:

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

输出:

           time  temp  foo
date                      
2014-10-1     1     3    0
2014-10-2     2     8    7
2014-10-3     3     4    9
2014-10-4     4     4    8
2014-10-5     5     6    2

期望输出:我正在使用此输出在 Highcharts 中,它要求它是一个像这样的列表字典

{'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]}
2个回答

38
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]}

3
我有点惊讶,这里没有针对此的to_json方法。 - Andy Hayden

0
创建一个每行包含字典的列表。
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)

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接