如何从python中的json数据中减去两个值

Question

我有以下 json 数据：

{
    "data": [
        {
            "id": "1",
            "type": "POK",
            "value": "34"
        },
        {
            "id": "1",
            "type": "TOK",
            "value": "32"
        },
        {
            "id": "1",
            "type": "POC",
            "value": "43"
        },
        {
            "id": "2",
            "type": "POK",
            "value": "45"
        },
        {
            "id": "2",
            "type": "TOK",
            "value": "67"
        },
        {
            "id": "3",
            "type": "POK",
            "value": "12"
        },
        {
            "id": "2",
            "type": "POC",
            "value": "23"
        },
        {
            "id": "3",
            "type": "TOK",
            "value": "34"
        },
        
    ]
}

在上面的数据中，我们可以看到id 为1, 2, 3，类型为POK TOK POC。我必须编写一个脚本来减去类型为POK 和TOK 的每个id 的值。例如，类型为POK 的id 1 从TOK 类型的id 1 中减去，所以34 - 32 得到答案2。 id 2 和 3 也是如此。

为此，我想到了首先在list 的dicts 中从上述json 中提取有用数据，然后减去每个值，如下所示：

import json


with open("data.json") as f:
    data = json.load(f)


l = []
for i in range(len(data['data'])):
    d = dict()
    if data['data'][i]['type'] == 'POK':
        d['id'] = data['data'][i]['id']
        d['value'] = data['data'][i]['value']
        d['type'] = 'POK'
        l.append(d)
    if data['data'][i]['type'] == 'TOK':
        d['id'] = data['data'][i]['id']
        d['timestamp'] = data['value'][i]['value']
        d['type'] = 'TOK'
        l.append(d)

现在在上面的代码之后，我有一个列表l，每个列表中有 6 个dict，其中包含数据

[
{'id': '1', 'value': '34', 'type': 'POK'}, 
{'id': '1', 'value': '32', 'type': 'TOK'}, 
{'id': '2', 'value': '45', 'type': 'POK'}, 
{'id': '2', 'value': '67', 'type': 'TOK'}, 
{'id': '3', 'value': '12', 'type': 'POK'}, 
{'id': '3', 'value': '34', 'type': 'TOK'}
]

在此之后，我无法理解如何减去类型为 'POK' 和 'TOK' 的每个 id 的值。谁能给我一些建议？有没有其他方法来处理这段代码？

Answer 1

以下是如何使用内置的all() 方法：

dct = [{'id': '1', 'value': '34', 'type': 'POK'},
       {'id': '1', 'value': '32', 'type': 'TOK'},
       {'id': '2', 'value': '45', 'type': 'POK'},
       {'id': '2', 'value': '67', 'type': 'TOK'},
       {'id': '3', 'value': '12', 'type': 'POK'},
       {'id': '3', 'value': '34', 'type': 'TOK'}]

for d1 in dct:
    for d2 in dct:
        if all([d1['id'] == d2['id'], d1['type'] == 'POK' ,d2['type'] == 'TOK']):
            dct.append({'id': d1['id'], 'value': str(int(d1['value'])-int(d2['value'])), 'type': 'subtracted'})

print(dct)

输出：

[{'id': '1', 'value': '34', 'type': 'POK'},
 {'id': '1', 'value': '32', 'type': 'TOK'},
 {'id': '2', 'value': '45', 'type': 'POK'},
 {'id': '2', 'value': '67', 'type': 'TOK'},
 {'id': '3', 'value': '12', 'type': 'POK'},
 {'id': '3', 'value': '34', 'type': 'TOK'},
 {'id': '1', 'value': '2', 'type': 'subtracted'},
 {'id': '2', 'value': '-22', 'type': 'subtracted'},
 {'id': '3', 'value': '-22', 'type': 'subtracted'}]

Answer 2

我不认为使用字典是组织减法数据的最直观方式。这是一个关于如何实现所需操作并将它们存储在列表中的示例：

list_of_dics = [
{'id': '1', 'value': '34', 'type': 'POK'}, 
{'id': '1', 'value': '32', 'type': 'TOK'}, 
{'id': '2', 'value': '45', 'type': 'POK'}, 
{'id': '2', 'value': '67', 'type': 'TOK'}, 
{'id': '3', 'value': '12', 'type': 'POK'}, 
{'id': '3', 'value': '34', 'type': 'TOK'}
]

TOKS = []
POKS = []
for entry in list_of_dics:
  if entry['type'] == 'TOK':
  ## We found a TOK store it for substraction
      TOKS.append((entry['id'], entry['value']))

  if entry['type'] == 'POK':
  ## We found a TOK store it for substraction
      POKS.append((entry['id'], entry['value']))

result = []

#I use this simple for loop since both lists should be of the same size as according to what you've shown on the answer.
for i in range(len(TOKS)):
  if TOKS[i][0] == POKS[i][0]:
  #We make the substractions since they have the same id
    result.append(int(POKS[i][1]) - int(TOKS[i][1]))

print(result)

Answer 3

看起来你的 for 循环除了再次构建相同的 json 对象外什么也没做。 相反，您可以直接遍历 json 并进行所需的计算。 工作示例：

import json

with open("data.json") as f:
    data = json.load(f)

result = {}
for d in data['data']:
    id = d['id']
    if d['type'] == 'POK':
        result[id] =  result.get(id, 0) + int(d['value'])
    if d['type'] == 'TOK':
        result[id] = result.get(id, 0) - int(d['value'])

print(result)

输出是

{'1': 2, '2': -22, '3': -22}