python - python中复杂的列表和字典查找

Question

我有一个元组列表和一个列表字典，如下所示。

# List of tuples
lot = [('Item 1', 43), ('Item 4', 82), ('Item 12', 33), ('Item 10', 21)]

# dict of lists
dol = {

    'item_category_one': ['Item 3', 'Item 4'],
    'item_category_two': ['Item 1'],
    'item_category_thr': ['Item 2', 'Item 21'],
}

现在我想查找 dol 中任何列表中的任何项目是否存在于 lot 中给出的任何元组中。如果满足此要求，那么我想向相应的元组添加另一个变量。

目前我正在这样做(看起来非常低效和丑陋)。我想知道实现此目标的最高效 和简洁 方法。有哪些可能性？

PS:我还希望在执行此操作时保留 lot 的顺序。

merged = [x[0] for x in lot]

for x in dol:
    for item in dol[x]:
        if item in merged:
            for x in lot:
                if x[0] == item:
                    lot[lot.index(x)] += (True, )

Answer 1

首先，在 dol 结构中构建一组所有值:

from itertools import chain
dol_values = set(chain.from_iterable(dol.itervalues()))

现在成员资格测试很有效，您可以使用列表理解:

[tup + (True,) if tup[0] in dol_values else tup for tup in lot]

演示:

>>> from itertools import chain
>>> dol_values = set(chain.from_iterable(dol.itervalues()))
>>> dol_values
set(['Item 3', 'Item 2', 'Item 1', 'Item 21', 'Item 4'])
>>> [tup + (True,) if tup[0] in dol_values else tup for tup in lot]
[('Item 1', 43, True), ('Item 4', 82, True), ('Item 12', 33), ('Item 10', 21)]