附加到Python字典中的列表

Question

有没有更优雅的方式来编写此代码？

我在做什么：我有键和日期。可能有多个日期分配给键，因此我正在创建一个包含日期列表的字典来表示这一点。以下代码可以正常工作，但是我希望有一个更优雅和Pythonic的方法。

dates_dict = dict() 
for key,  date in cur:
    if key in dates_dict:
        dates_dict[key].append(date)
    else:
        dates_dict[key] = [date]

我期望下面的工作，但我不断得到一个NoneType没有属性附加错误。

dates_dict = dict() 
for key,  date in cur:
    dates_dict[key] = dates_dict.get(key, []).append(date)

这可能与以下事实有关：

print([].append(1)) 
None

但为什么？

Answer 1

list.append返回None，因为它是就地操作，并且您将其分配回dates_dict[key]。因此，下一次dates_dict.get(key, []).append您实际要做的是None.append。这就是为什么它失败了。相反，您只需

dates_dict.setdefault(key, []).append(date)

但是，我们collections.defaultdict仅出于此目的。你可以做这样的事情

from collections import defaultdict
dates_dict = defaultdict(list)
for key, date in cur:
    dates_dict[key].append(date)

如果key在字典中找不到，则会创建一个新的列表对象。

注意：defaultdict如果在字典中找不到关键字，则由于会创建一个新列表，因此会产生意想不到的副作用。例如，如果您只想检索键的值（该值不存在），它将创建一个新列表并返回它。