python - 将 pandas 数据框转换为自定义 json

Question

我有一个具有以下结构的数据框:

import pandas
df = pandas.DataFrame({
     "date" : ["day_1","day_1","day_2","day_3","day_4"],
     "block" : ["1", "2", "1", "1", "1"],
     "start_time" : ["08:00","11:00","08:00","09:00", "09:00"],
     "end_time" :["10:00","15:00","10:00","11:00", "13:00"]
})

如何将其转换为具有以下结构的 json:

 [
 {
  "day_1":[
     {
        "1":{
           "start_time":"08:00:00",
           "end_time":"10:00:00"
        },
        "2":{
           "start_time":"11:00:00",
           "end_time":"15:00:00"
        }
     }
  ],
  "day_2":{
     "1":{
        "start_time":"2019-04-29 08:00:00",
        "end_time":"2019-04-29 10:00:00"
     }
  }
 }
 ]

区 block day_n的所有值都应写入json中的day_n下。

Answer 1

我认为这段代码会产生您想要的输出:

import json
init_dict = {date: {} for date in df['date'].unique()}
for ii in init_dict.keys():
    for jj in df.loc[df['date']==ii, 'block']:
        init_dict[ii][jj] = {'start_time': df.loc[(df['date']==ii) & (df['block']==jj), 'start_time'].values[0], 'end_time': df.loc[(df['date']==ii) & (df['block']==jj), 'end_time'].values[0]}

json_out = json.dumps(init_dict)

请告诉我这是否是您想要的。