删除json中不必要的元素

Question

我有JSON数据，我想在其中删除不必要的元素。我使用的是element.pop()，这是我们正常的做法。但我想知道，如果我在JSON中有数千个元素，而我的需求仅限于element0、element1和element6，那么我需要再次使用element.pop()。

我们有什么东西吗，在那里我不需要弹出所有不必要的元素？因为，如果我的要求只有3个元素，那么我需要pop个数千个元素。这需要时间。

data.json

[
{
"fields": {
"element0": "val0",
"element1": "val1",
"element2": "val2",
"element3": "val3",
"element4": "val4",
"element5": "val5",
"element6": "val6"
}
},
{
"fields": {
"element0": "val0",
"element1": "val1",
"element2": "val2",
"element3": "val3",
"element4": "val4",
"element5": "val5",
"element6": "val6"
}
}
]

python代码

import json
with open('data.json','r') as f:
s = f.read()
data = json.loads(s)
### my expectation
for element in data:
element['fields'].pop('element2', None)
element['fields'].pop('element3', None)
element['fields'].pop('element4', None)
element['fields'].pop('element5', None)

预期.json

[
{
"fields": {
"element0": "val0",
"element1": "val1",
"element6": "val6"
}
},
{
"fields": {
"element0": "val0",
"element1": "val1",
"element6": "val6"
}
}
]

Answer 1

如果您只需要data:中的某些元素，您可以执行以下操作

>>> data = [{'fields': {'element0': 'val0',
...    'element1': 'val1',
...    'element2': 'val2',
...    'element3': 'val3',
...    'element4': 'val4',
...    'element5': 'val5',
...    'element6': 'val6'}},
...  {'fields': {'element0': 'val0',
...    'element1': 'val1',
...    'element2': 'val2',
...    'element3': 'val3',
...    'element4': 'val4',
...    'element5': 'val5',
...    'element6': 'val6'}}]

然后只需选择您需要的字段，然后从嵌套结构中抓取：

>>> NEEDED = "element0", "element1", "element6"
>>> for element in data:
...     fields = element['fields']
...     element['fields'] = {n: fields.get(n) for n in NEEDED}
...

数据将为：

>>> data
[{'fields': {'element0': 'val0', 'element1': 'val1', 'element6': 'val6'}}, {'fields': {'element0': 'val0', 'element1': 'val1', 'element6': 'val6'}}]

或者只是重新创建数据：

>>> NEEDED = "element0", "element1", "element6"
>>> data2 = [{'fields': {n: ele['fields'].get(n) for n in NEEDED}} for ele in data]