按照特定模式打印 json

Question

我有一个这样的文件.csv：

ID;Name;Sports
1;Mark;["Football", "Volleyball"]
2;Luuk;["Fencing"]
3;Carl;["Swimming", "Basketball"]

我的目标是在文件.json 上打印以下内容：

[
{
"ID": 1,
"Name": "Mark",
"Sports": ["Football", "Volleyball"]
}
,
{
"ID": 2,
"Name": "Luuk",
"Sports": ["Fencing"]
}
,
{
"ID": 3,
"Name": "Carl",
"Sports": ["Swimming", "Basketball"]
}
]

我生成列表的代码：

from ast import literal_eval
import pandas as pd

df = pd.read_csv("test.csv", sep=";", converters={'Sports': literal_eval})

columns = list(df.columns)

json = []
for i in range(len(df)):
    line = {}
    for j in range(len(columns)):
        line[columns[j]] = df.iloc[i, j]
    json.append(line)

如何以上述相同格式巧妙地打印我的列表？

Answer 1

使用DataFrame.to_json:

from ast import literal_eval

df = pd.read_csv("test.csv", sep=";", converters={'Sports': literal_eval})


json = df.to_json(orient='records')
print (json)
[{"ID":1,"Name":"Mark","Sports":["Football","Volleyball"]},
 {"ID":2,"Name":"Luuk","Sports":["Fencing"]},
 {"ID":3,"Name":"Carl","Sports":["Swimming","Basketball"]}]

编辑：如果需要创建json 文件使用DataFrame.to_dict：

from ast import literal_eval

df = pd.read_csv("test.csv", sep=";", converters={'Sports': literal_eval})

d = df.to_dict(orient='records')
print (d)
[{"ID":1,"Name":"Mark","Sports":["Football","Volleyball"]},
 {"ID":2,"Name":"Luuk","Sports":["Fencing"]},
 {"ID":3,"Name":"Carl","Sports":["Swimming","Basketball"]}]

import json
with open('result.json', 'w') as fp:
    json.dump(d, fp)

Answer 2

import json

data = {"data" : []}

for index, row in df.iterrows():
    data.append(row.to_dict())

with open('data.json', 'w') as fp:
    json.dump(data, fp)