Python - 枚举过滤字典的非笨拙方式

Question

我有这样的代码（ d是一些字典， s是一些集合）：

for i, (k, v) in enumerate((k, v) for k, v in d.items() if k in s):
    ...

即使我们忽略丑陋的重复，所有文字实际上都有相当长的长度，因此重复k和v 3 次不是一种选择。 最好的写法是什么？ 现在，我们使用以下（也是丑陋的）选项：

i = -1
for k, v in d.items():
    if k not in s:
        continue
    i += 1
    ...

另外的选择：

for i, (k, v) in enumerate((k, d[k]) for k in d if k in s):
    ...

，但它基本上具有相同的重复次数。

我可以使用默认提供的任何库（即没有pip install ）。

Answer 1

使用额外的一行，但复杂性要低得多，您可以执行以下操作：

for i, k in enumerate(filter(s.__contains__, d)):
    v = d[k]

这种方法的一种变体是使用集合交集（虽然这可能会扰乱您的插入顺序）：

for i, k in enumerate(s & d.keys()):
    v = d[k]

Answer 2

dict理解是过滤dict的最快方法：

filtered_dict = {key: value for key, value in d.items() if key in s}

因此，我建议定义一个任务 function 来完成项目的工作并在这个字典理解中做所有事情：

def task(key, value):
    # do the required job on the dict items
    ....
    return result


{key: task(key, value) for key, value in d.items() if key in s}

结果是一个字典，其中包含每个键/值项的计算结果。 当然，如果额外需要索引，您可以添加enumerate 。