组合具有特定匹配值的字典列表

Question

给定python中的a_list和b_list，我想合并它们，使结果为c_list。

在下面的例子中，我只想合并与'code'键' a '匹配的元素。

a_list = [{'code': 'A', 'foo_sum': 1}, {'code': 'B', 'foo_sum': 2}]
b_list = [{'code': 'A', 'bar_sum': 3}]
c_list = [{'code': 'A', 'foo_sum': 1, 'bar_sum': 3}, {'code': 'B', 'foo_sum': 2}]

我想使用numpy，但不能使用它，因为它不一定匹配元素的长度和顺序。

并且，我试图在迭代过程中确定这一点，但无法实现c，因为局部变量在推导符号中不可用。

for a in b_list:
if any(b["code"] == a["code"] for i, b in enumerate(b_list):
# want to use i variable, but not dified.
a["bar_sum"] = b_list[i]["code"]

Answer 1

另一个解决方案:

a_list = [{"code": "A", "foo_sum": 1}, {"code": "B", "foo_sum": 2}]
b_list = [{"code": "A", "bar_sum": 3}]

out = {}
for d in a_list + b_list:
out[d["code"]] = {**out.get(d["code"], {}), **d}
print(list(out.values()))

打印:

[{"code": "A", "foo_sum": 1, "bar_sum": 3}, {"code": "B", "foo_sum": 2}]

Answer 2

你可以试试

1. 合并两个列表
2. 按code键排序，再按code键排序
3. 将组内所有字典合并为一个字典

from itertools import groupby
from collections import ChainMap
c_list = a_list + b_list
c_list = sorted(c_list, key=lambda d: d.get('code'))
res = [dict(ChainMap(*group)) for key, group in groupby(c_list, lambda d: d.get('code'))]

print(res)
[{'code': 'A', 'bar_sum': 3, 'foo_sum': 1}, {'code': 'B', 'foo_sum': 2}]