首页 > 解决方案 > 我需要将所有熊猫行数据保存到 JSON 中的特定字段

问题描述

我已经用熊猫对数据进行了一些处理。现在我在 pandas dataFrame 对象中有它。我需要保存在 JSON 中。但我的要求是每一行都应该在名为“项目”的字段中。例如。

[{
  'item': row_1
},
{
  'item': row_2
},{
  'item': row_2
}]

row_1 是 {'column1':'data11', 'column2':'data12'}

标签: pythonpandas

解决方案


您可以在数据帧上使用 to_dict('records') 方法来获取记录列表。假设您有两列和三行(如上所示),生成的记录列表将如下所示

[{'column1': 'data11', 'column2': 'data12'}, 
 {'column1': 'data21', 'column2': 'data22'}, 
 {'column1': 'data31', 'column2': 'data32'}]

然后,您可以遍历此 dicts 列表以创建具有所需格式的列表。总而言之,这看起来像

# get the list of records from the original dataframe df
record_list = df.to_dict('records')

# use a list comprehension to get the record list in the desired format
formatted_list = [{'item': record} for record in record_list]

推荐阅读