问题描述
附加字典键的条件列表:
results = {}
for item,data in myDict.items():
if (key := data['attr']) not in results:
results[key] = []
results[key].append(item)
理想的词典理解:
results = {key: results[key].append(item) for item,data in myDict.items()}
但很明显,如果键没有被实例化为空列表,它就不会工作。
同样,results[key].append(item)
也不起作用。
解决方法
你不能把它写成一个推导式:推导式的每个元素都是你正在迭代的东西的过滤版本。理解中的所有元素都被独立处理并附加到单个容器中。您要求根据密钥附加到不同的容器,因此无法理解。
我能想到的最简单的方法是使用 collections.defaultdict
:
from collections import defaultdict
result = defaultdict(list)
for item,data in myDict.items():
result[data['attr']].append(item)
现在您可以通过创建一个具有一些无关副作用的函数来欺骗系统并使用它来过滤输出。在这种情况下,唯一真正的好处是将已经清晰易读的 for
循环写在一行中。您可能希望在不实际存储函数结果的情况下使用推导式:
from collections import deque,defaultdict
results = defaultdict(list)
deque((results[data['attr']].append(item) for item,data in myDict.items()),maxlen=0)
这个将 collections.deque
与 maxlen=0
结合使用的技巧是纯粹为了副作用而使用迭代器的更快方法之一。