Python 不确定如何操作子词典

我正在使用Twitch API，并最终得到了以下词典：

用户数据

{"data": [
    {"display_name": "John", "id": "123"}, 
    {"display_name": "Frank", "id": "456"}, 
    {"display_name": "Billy", "id": "789"}]}

流数据

{"data": [
    {"id": "333444", "user_id": "456", "title": "Franks Stream"}
    {"id": "555666", "user_id": "789", "title": "Billys Stream"}
    {"id": "111222", "user_id": "123", "title": "Johns Stream"}]}

我不能保证两者的顺序相同，因此我希望使用id/user\u id将两者结合起来：

组合数据

{"data": [
    {"id": "333444", "user_id": "456", "title": "Franks Stream", "display_name": "Frank"}
    {"id": "555666", "user_id": "789", "title": "Billys Stream", "display_name": "Billy"}
    {"id": "111222", "user_id": "123", "title": "Johns Stream", "display_name": "John"}]}

---

如果可能的话，我想合并更多的字段，但是可以这样加入这些词典吗？

您可以使用

itertools。groupby

：

import itertools
from functools import reduce
d = {'data': [{'display_name': 'John', 'id': '123'}, {'display_name': 'Frank', 'id': '456'}, {'display_name': 'Billy', 'id': '789'}]}
d1 = {'data': [{'id': '333444', 'user_id': '456', 'title': 'Franks Stream'}, {'id': '555666', 'user_id': '789', 'title': 'Billys Stream'}, {'id': '111222', 'user_id': '123', 'title': 'Johns Stream'}]}
combined = sorted(d['data']+d1['data'], key=lambda x:x.get('user_id', x['id']))
new_data = [list(b) for _, b in itertools.groupby(combined, key=lambda x:x.get('user_id', x['id']))]
final_results = {'data':[reduce(lambda x, y:{**x, **y}, i) for i in new_data]}

输出：

{'data': [
  {'display_name': 'John', 'id': '111222', 'user_id': '123', 'title': 'Johns Stream'}, 
  {'display_name': 'Frank', 'id': '333444', 'user_id': '456', 'title': 'Franks Stream'}, 
  {'display_name': 'Billy', 'id': '555666', 'user_id': '789', 'title': 'Billys Stream'}
]}

---

让我们在这里使用

pandas

import pandas as pd 
df2=pd.DataFrame(d2['data'])
df1=pd.DataFrame(d1['data']).rename(columns={'id':'user_id'})
{'data':df1.merge(df2).to_dict('r')}
Out[150]: 
{'data': [{'display_name': 'John',
   'id': '111222',
   'title': 'Johns Stream',
   'user_id': '123'},
  {'display_name': 'Frank',
   'id': '333444',
   'title': 'Franks Stream',
   'user_id': '456'},
  {'display_name': 'Billy',
   'id': '555666',
   'title': 'Billys Stream',
   'user_id': '789'}]}

---

您可以先将用户数据dict转换为

id

-到-

显示\u名称

映射：

u = {"data": [
    {"display_name": "John", "id": "123"},
    {"display_name": "Frank", "id": "456"},
    {"display_name": "Billy", "id": "789"}]}
s = {"data": [
    {"id": "333444", "user_id": "456", "title": "Franks Stream"},
    {"id": "555666", "user_id": "789", "title": "Billys Stream"},
    {"id": "111222", "user_id": "123", "title": "Johns Stream"}]}
m = dict(reversed(tuple(i.values())) for i in u['data'])
combined = {'data': [{**i, 'display_name': m[i['user_id']]} for i in s['data']]}

组合后

变成：

{'data': [{'id': '333444', 'user_id': '456', 'title': 'Franks Stream', 'display_name': 'Frank'}, {'id': '555666', 'user_id': '789', 'title': 'Billys Stream', 'display_name': 'Billy'}, {'id': '111222', 'user_id': '123', 'title': 'Johns Stream', 'display_name': 'John'}]}

设置

user_data = {
    "data": [
        {"display_name": "John", "id": "123"}, 
        {"display_name": "Frank", "id": "456"}, 
        {"display_name": "Billy", "id": "789"}]}

stream_data = {
    "data": [
        {"id": "333444", "user_id": "456", "title": "Franks Stream"},
        {"id": "555666", "user_id": "789", "title": "Billys Stream"},
        {"id": "111222", "user_id": "123", "title": "Johns Stream"}]}

解决方案

使用字典理解，您可以创建键入

id

字段的用户数据的新字典，然后使用此数据更新流数据

new_user_data = {row.get("id"): row.get('display_name')
                 for row in user_data['data']}
>>> new_user_data
{'123': 'John', '456': 'Frank', '789': 'Billy'}

现在，使用此词典更新数据中的

显示名称

（如果不想改变原始数据，您可能希望复制一份）

如果您有多个用户数据要存储为值（即不仅仅是

id

作为键和

display\u name

作为值），您可以使用保存它们

user_data = {
    "data": [
        {"display_name": "John", "id": "123", "description": "boring"}, 
        {"display_name": "Frank", "id": "456", "description": "smart"}, 
        {"display_name": "Billy", "id": "789", "description": "funny"}]}


from collections import namedtuple

UserData = namedtuple('UserData', ('display_name', 'description'))

for row in user_data['data']:
    id_ = row.pop('id')
    new_user_data[id_] = UserData(**row)

for row in stream_data['data']:
    id_ = row.get('user_id')
    row.update(**new_user_data[id_].__dict__)


>>> new_user_data
{'123': UserData(display_name='John', description='boring'),
 '456': UserData(display_name='Frank', description='smart'),
 '789': UserData(display_name='Billy', description='funny')}

>>> stream_data
{'data': [
    {'description': 'smart',
     'display_name': 'Frank',
     'id': '333444',
     'title': 'Franks Stream',
     'user_id': '456'},
    {'description': 'funny',
     'display_name': 'Billy',
     'id': '555666',
     'title': 'Billys Stream',
     'user_id': '789'},
    {'description': 'boring',
     'display_name': 'John',
     'id': '111222',
     'title': 'Johns Stream',
     'user_id': '123'}]}

---

您可能需要检查Pandash如何使用“id”键进行组合这两个字典中的Id必须相同，对吗？虽然我喜欢将此作为问题的解决方案，但如何将更多字段从用户数据添加到拖缆数据？假设我想要[“数据”][“描述”]？

user_data = {
    "data": [
        {"display_name": "John", "id": "123", "description": "boring"}, 
        {"display_name": "Frank", "id": "456", "description": "smart"}, 
        {"display_name": "Billy", "id": "789", "description": "funny"}]}


from collections import namedtuple

UserData = namedtuple('UserData', ('display_name', 'description'))

for row in user_data['data']:
    id_ = row.pop('id')
    new_user_data[id_] = UserData(**row)

for row in stream_data['data']:
    id_ = row.get('user_id')
    row.update(**new_user_data[id_].__dict__)


>>> new_user_data
{'123': UserData(display_name='John', description='boring'),
 '456': UserData(display_name='Frank', description='smart'),
 '789': UserData(display_name='Billy', description='funny')}

>>> stream_data
{'data': [
    {'description': 'smart',
     'display_name': 'Frank',
     'id': '333444',
     'title': 'Franks Stream',
     'user_id': '456'},
    {'description': 'funny',
     'display_name': 'Billy',
     'id': '555666',
     'title': 'Billys Stream',
     'user_id': '789'},
    {'description': 'boring',
     'display_name': 'John',
     'id': '111222',
     'title': 'Johns Stream',
     'user_id': '123'}]}