python - 从列表中获取相关词典

Question

我有两个不同词典的列表(ListA 和 ListB)。

listA 中的所有字典都有字段“id”和“external_id” listB 中的所有词典都有字段“num”和“external_num”

我需要获取所有字典对，其中 external_id 的值 = num 和 external_num 的值 = id.

我可以使用这段代码实现:

for dictA in ListA:
    for dictB in ListB:
        if dictA["id"] == dictB["external_num"] and dictA["external_id"] == dictB["num"]:

但我看到了很多漂亮的 python 表达式，我想有可能得到更 pythonic 风格的结果，不是吗？

我喜欢:

res = [A, B for A, B in listA, listB if A['id'] == B['extnum'] and A['ext'] == B['num']]

Answer 1

你已经很接近了，但你并没有告诉 Python 你想如何连接两个列表来获得字典对 A 和 B。

如果你想比较ListA中的所有字典和ListB中的所有字典，你需要itertools.product :

from itertools import product

res = [A, B for A, B in product(ListA, ListB) if ...]

或者，如果您想要相同索引的对，请使用 zip :

res = [A, B for A, B in zip(ListA, ListB) if ...]

---

如果您不需要一次构建整个列表，请注意您可以使用 itertools.ifilter选择你想要的对:

from itertools import ifilter, product

for A, B in ifilter(lambda (A, B): ..., 
                    product(ListA, ListB)):
    # do whatever you want with A and B

(如果您使用 zip 执行此操作，请改用 itertools.izip 以最大化性能)。

---

关于 Python 3.x 的注释:

• zip 和 filter no longer return lists ，因此 itertools.izip 和 itertools.ifilter 不再存在(就像 range 推出了 xrange)和您只需要 itertools 中的 product；和
• lambda (A, B): 是 no longer valid syntax ;您将需要编写过滤函数以采用单个元组参数 lambda t: 和例如将 A 替换为 t[0]。