如何将数据框转换为每行具有相似索引的 jsonl？

Question

我有dataframe，它有“输入”作为所有行的索引。数以千计。

  df1 = 
  index   item name   item quantity
  input       apple         4
  input       orange        3
  input       lemon         6  

我需要把它变成一个像这样的jsonl 文件。由于 Shopify GraphQL Admin API 批量导入的要求，我需要它看起来像这样。

{ "input": { "item name": "apple", "item quantity": "4"}}
{ "input": { "item name": "orange", "item quantity": "3"}}
{ "input": { "item name": "lemon", "item quantity": "6"}}

我可以使用df1.to_json(orient='records', lines=True) 将其转换为jsonl，而不需要索引看起来像这样

{ "item name": "apple", "item quantity": "4"}}
{ "item name": "orange", "item quantity": "3"}}
{ "item name": "lemon", "item quantity": "6"}}

但我需要将“输入”键添加到前面，但我不知道该怎么做。我是 json 新手。

我也尝试了df1.to_json(orient="index")，但它给了我一个错误ValueError: DataFrame index must be unique for orient='index'，表明每个索引都必须是唯一的。

感谢任何帮助。干杯

Answer 1

您需要的输出是：

{ "input": { "item name": "apple", "item quantity": "4"}}
{ "input": { "item name": "orange", "item quantity": "3"}}
{ "input": { "item name": "lemon", "item quantity": "6"}}

orient='index' 在这种情况下不起作用，因为索引不是唯一的。上面的答案看起来很棒。您的预期输出是单独的 JSON。

这是一种解决方法，如果您本身需要相同的格式，请根据您所需的格式

import pandas as pd
import json
data=[['apple',4],['orange','3'],['lemon',6]]
df=pd.DataFrame(data,columns=['item name','item quantity'])
json_as_str=df.to_json(orient="index")
json_value=json.loads(json_as_str)
string_formatted=[]
for key,val in json_value.items():
    string_formatted.append("{'input':%s}" %val)
with open("file_name_here.jsonl","a") as fh:
    for i in string_formatted:
        i=i.replace("'",'"')
        fh.write(f"{i}\n")

输出：

{"input":{"item name": "apple", "item quantity": 4}}
{"input":{"item name": "orange", "item quantity": 3}}
{"input":{"item name": "lemon", "item quantity": 6}}