我当前加载的文件如下:
import ndjson
with open('full_simplified_arm.ndjson') as f:
data = ndjson.load(f)
然而,我想用Python将这个文件转换成csv,并想知道如何做到这一点?感谢
您可以使用csv.DictWriter
来完成此操作。
用ndjson.load
加载数据,以这种形式存储。
[{'id': 1, 'name': 'Alice'}, {'id': 2, 'name': 'Bob'}, {'id': 3, 'name': 'Carol'}]
您可以将列表中第一项中的键作为表的字段名。
fields = data[0].keys()
然后定义一个csv.DictWriter
来写入标题和行。
writer = csv.DictWriter(f, fieldnames=fields)
writer.writeheader()
for item in data:
writer.writerow(item)
完整解决方案:
import csv
import ndjson
with open('./users.ndjson', 'r') as f:
data = ndjson.load(f)
with open('./users.csv', 'w', newline='') as f:
# I'm assuming that each item has the same keys
fields = data[0].keys()
writer = csv.DictWriter(f, fieldnames=fields)
writer.writeheader()
for item in data:
writer.writerow(item)
输入
{"id":1,"name":"Alice"}
{"id":2,"name":"Bob"}
{"id":3,"name":"Carol"}
输出
id,name
1,Alice
2,Bob
3,Carol
注:样本数据是在线检索的