在字典中添加新列表或附加到列表（如果存在）是不是有一个很好的习语？

Posted 2023-02-22

【中文标题】在字典中添加新列表或附加到列表（如果存在）是不是有一个很好的习语？

【英文标题】：Is there a nice idiom for adding a new list or appending to a list (if present) in a dictionary?在字典中添加新列表或附加到列表（如果存在）是否有一个很好的习语？

【发布时间】：2012-03-07 10:40:27

【问题描述】：

这是我第二次实现这样的东西，我怀疑必须有更好的（阅读：更多pythonic）方法来做到这一点：

phone_book = 

def add_number(name,number):
  if name in phone_book:
    phone_book['name'].append(number)
  else:
    phone_book['name'] = [number]

我意识到使用条件赋值可能会使代码更简洁，但我怀疑可能有更好的方法来解决这个问题。我对只缩短代码不感兴趣。

【参考方案1】：

是的，您可以使用defaultdict。有了这个dict 子类，当你访问字典中的一个元素时，如果一个值不存在，它会使用你指定的构造函数自动创建一个。

from collections import defaultdict
phone_book = defaultdict(list)

def add_number(name, number):
    phone_book[name].append(number)

【参考方案2】：

像这样使用dict的setdefault：

phone_book.setdefault('name', []).append(number)