Python:list()作为字典的默认值

Question

我有Python代码,看起来像:

if key in dict:
  dict[key].append(some_value)
else:
  dict[key] = [some_value]

但我认为应该有一些方法可以绕过这个'if'语句.我试过了

dict.setdefault(key, [])
dict[key].append(some_value)

和

dict[key] = dict.get(key, []).append(some_value)

但两人都抱怨"TypeError:unhashable type:'list'".有什么建议？谢谢!

Answer 1

最好的方法是使用collections.defaultdict一个list默认值:

from collections import defaultdict
dct = defaultdict(list)

然后使用:

dct[key].append(some_value)

如果密钥尚未出现在映射中,字典将为您创建一个新列表.collections.defaultdict是一个子类,dict并且行为与普通dict对象一样.

使用标准时dict,dict.setdefault()正确设置dct[key]为默认值,以便版本应该正常工作.你可以用以下方式链接该呼叫.append():

>>> dct = {}
>>> dct.setdefault('foo', []).append('bar')  # returns None!
>>> dct
{'foo': ['bar']}

但是,通过使用dct[key] = dct.get(...).append()您将值替换为dct[key]输出.append(),即None.