如何在 python 中将字典附加到 JSON 文件？答案

【问题标题】：How do i append a dictionary to a JSON file in python?如何在 python 中将字典附加到 JSON 文件？
【发布时间】：2021-08-28 08:25:30
【问题描述】：

我的 JSON 看起来像这样：

{'data': [], 'directed': False, 'multigraph': False, 'elements': {'nodes': [{'data': {'id': 'B2', 'value': 'B2', 'name': 'B2'}}, {'data': {'id': 'SCHROEDER PLZ', 'value': 'SCHROEDER PLZ', 'name': 'SCHROEDER PLZ'}}, {'data': {'id': 'D4', 'value': 'D4', 'name': 'D4'}}, {'data': {'id': 'BLAB PLZ', 'value': 'BLAB PLZ', 'name': 'BLAB PLZ'}}], 'edges': [{'data': {'source': 'B2', 'target': 'SCHROEDER PLZ'}}, {'data': {'source': 'D4', 'target': 'BLAB PLZ'}}]}}

JSON 是我的代码中“加载”的结果：

import pandas as pd
import networkx as nx
import json

df= pd.read_csv('.../graph.csv')
g = nx.from_pandas_edgelist(df, source='DISTRICT', target='STREET')
x = nx.cytoscape_data(g)
dump = json.dumps(x)
loads = json.loads(dump)

这是我的 csv 文件结构：第一条记录是字段名。

OFFENSE_DESCRIPTION,DISTRICT,DAY_OF_WEEK,STREET,INCIDENT_NUMBER,size
INVESTIGATE PERSON,B2,Thursday,SCHROEDER PLZ,854652314,10
INVESTIGATE PERSON,D4,Friday,BLAB PLZ,457856954,3

我想附加位于我的 csv 文件中的“大小”值。

其实结果一定是像下面的JSON。在“节点”标签中，在“数据”中我想添加“大小”字段值。

 {'data': [], 'directed': False, 'multigraph': False, 'elements': {'nodes': [{'data': {'id': 'B2', 'value': 'B2', 'name': 'B2','size':10}}, {'data': {'id': 'SCHROEDER PLZ', 'value': 'SCHROEDER PLZ', 'name': 'SCHROEDER PLZ','size':10}}, {'data': {'id': 'D4', 'value': 'D4', 'name': 'D4','size':3}}, {'data': {'id': 'BLAB PLZ', 'value': 'BLAB PLZ', 'name': 'BLAB PLZ','size':3}}], 'edges': [{'data': {'source': 'B2', 'target': 'SCHROEDER PLZ'}}, {'data': {'source': 'D4', 'target': 'BLAB PLZ'}}]}}

【问题讨论】：

readcsv 和 df 具有相同的值。可以直接写df = pd.read_csv('.../graph.csv')
@Corralien 谢谢。我编辑了我的代码。

标签： python json python-3.x pandas networkx

【解决方案1】：

一个优雅的解决方案是更新 networkx 中的节点属性而不是输出字典。使用nx.set_node_attributes:

df = pd.read_csv('.../graph.csv')
size = dict(df[['DISTRICT', 'size']].values.tolist()
            + df[['STREET', 'size']].values.tolist())

g = nx.from_pandas_edgelist(df, source='DISTRICT', target='STREET')
nx.set_node_attributes(g, size, 'size')
x = nx.cytoscape_data(g)

>>> print(json.dumps(x['elements']['nodes'], indent=4))
[
    {
        "data": {
            "size": 10,
            "id": "B2",
            "value": "B2",
            "name": "B2"
        }
    },
    {
        "data": {
            "size": 10,
            "id": "SCHROEDER PLZ",
            "value": "SCHROEDER PLZ",
            "name": "SCHROEDER PLZ"
        }
    },
    {
        "data": {
            "size": 3,
            "id": "D4",
            "value": "D4",
            "name": "D4"
        }
    },
    {
        "data": {
            "size": 3,
            "id": "BLAB PLZ",
            "value": "BLAB PLZ",
            "name": "BLAB PLZ"
        }
    }
]

【讨论】：

@Coralien。谢谢你的回答。