字典推导式中检查键存在：能否用推导式实现元素分组？

嘿，这个问题问得挺有意思的！首先得搞明白你原来的字典推导式为啥不管用——其实核心问题出在字典推导式的执行逻辑和append()的返回值上。

你预先声明了d = {}，但在推导式里访问的d始终是那个初始的空字典，因为推导式是先完整计算所有键值对，最后才把结果赋值给d的。也就是说，在推导过程中，正在构建的新字典还没覆盖原来的d，所以每次判断num not in d时，d都是空的，所有元素都会走[num]的分支。就算假设能触发else分支，d[num].append(num)的返回值是None，这会导致键对应的值变成None，完全不是你想要的结果。

那能不能用字典推导式实现呢？当然可以，不过得换个思路，不能依赖推导过程中修改字典本身。这里有两种常用方法：

方法一：遍历唯一元素，生成对应列表 #

直接针对每个唯一元素，筛选出原列表里所有相同的元素：

l = [1, 2, 2, 3, 4, 4, 1, 1]
d = {num: [x for x in l if x == num] for num in set(l)}

这种写法简单直观，结果完全符合你的期望。缺点是效率稍低——每个唯一元素都要遍历一遍整个列表，列表很大的话会有O(n²)的时间复杂度。

方法二：先统计次数，再生成重复列表 #

用collections.Counter先统计每个元素的出现次数，再用推导式生成对应长度的列表：

from collections import Counter
l = [1, 2, 2, 3, 4, 4, 1, 1]
counts = Counter(l)
d = {num: [num] * counts[num] for num in counts}

这个方法效率更高，Counter统计是O(n)，推导式是O(k)（k是唯一元素的数量），整体是O(n)的时间复杂度，适合处理大列表。

不过要注意，字典推导式本身是一次性构建所有键值对的工具，不像循环那样可以逐步修改字典状态。所以如果追求极致效率，defaultdict或者普通的for循环（比如for num in l: d.setdefault(num, []).append(num)）其实是更直接的选择，但既然你好奇字典推导式的实现方式，上面两种都能满足需求~

内容的提问来源于stack exchange，提问作者pstatix