使用python从csv创建自定义json文件

Question

我有一个包含 3 个字段的 csv 文件。我想根据表格自定义格式创建一个 json 文件。我想使用dash cytoscape中的json文件。

我曾尝试使用 pandas 作为数据框，并将数据框更改为 json 格式。我看到了this question。由于不同的 json 格式不同，这个问题我无法使用答案。 CSV 表：

DISTRICT,MONTH
B2,2
A7,5
A1,9

...

我想把表格转成json格式如下：

[           {'data': {'id': 'B2', 'label': 'B2'}},
            {'data': {'id': 'A7', 'label': 'A7'}},
            {'data': {'id': 'A1', 'label': 'A1'}},
            {'data': {'id': '2', 'label': '2'}},
            {'data': {'id': '5', 'label': '5'}},
            {'data': {'id': '9', 'label': '9'}},

            {'data': {'source': 'B2', 'target': '2'}},
            {'data': {'source': 'A7', 'target': '5'}},
            {'data': {'source': 'A1', 'target': '9'}}
        ]

Answer 1

这些代码应该可以完成这项工作。如果您对数据进行一些花哨的分析，Pandas 可能会更容易一些。我不是粉丝（因为它经常让我陷入不必要的技术细节，加上版本混乱），所以我坚持使用 Python 内置包。

import csv
import json

with open('yourcsv.csv') as csvfile:
    labels = []
    mappings = []
    csvreader = csv.reader(csvfile, delimiter=',')
    # skips your header
    next(csvreader)
    for row in csvreader:
        labels.append({
            'data': {'id': row[0], 'label': row[0]},
            'data': {'id': row[1], 'label': row[1]}
        })
        mappings.append({
            'data': {'source': row[0], 'target': row[1]}
        })

with open('yourjson.json', 'w') as f:
    # indent does the prettify
    json.dump(labels + mappings, f, indent=4)