将python数据帧转换为json格式

Question

我正在尝试将 python 字典转换为 json 。
输入数据框如下所示：

user_id |last_update_time |states| tile                       | update_time | message
11       |1571737828      |S231  |{'Action1': [200, 250, 300],|1571737828   | traffic
                                  'Action2': [150, 200, 400]}
22       |1571737828      |S450  |{'Action1': [100, 150, 400],|1571737828   | traffic
                                  'Action2': [350, 500, 700]}

已经尝试过了，但是它没有提供所需的输出格式，如下所示的预期json输出格式

adf = input_df.groupby('user_id').apply(lambda x: x.set_index('message').to_dict(orient='index')).to_dict()   

预期的 json 输出文件：

{
  11:{ 
           "states": "s123", "last_update_time":111342342342,
           "tile": {
               "Action1": [200, 250, 300],
               "Action2": [150, 200, 400]
                      }, "update_time":111342342342
           },
  22:{ 
           "states": "s124", "last_update_time":1113423442342,
           "tile": {{
               "Action1": [100, 150, 400],
               "Action2": [350, 500, 700]
                      }}, "update_time":1141342342342
           },
"message":"traffic"
}

这个输出的解决方案对我很有帮助。

Answer 1

假设id_user 是您的数据框的索引，以下行应该会给您预期的结果：

df.to_json("/path/to/myjson.json", orient="index")

Answer 2

我会尝试：

# you set the the user_id as index
adf=df.set_index('user_id')

# you convert the dataframe to a dict
# keys are dataframe indexes (user_id)
adf.to_dict(orient='index')

输出是期望的。