python - 向集合字典中添加元素

Question

我有一个字典 d 集合。在某些键 key 下添加元素时，我考虑了两种情况 key in d 和 not key in d:

from random import randint

for i in range(1000):
    key = randint(0, 100)

    if key in d:
        d[key] |= set([randint(0, 10)])
    else:
        d[key] = set([randint(0, 10])

有没有更简洁的方法来做到这一点？我可以统一这两种情况吗？

Answer 1

我喜欢使用 defaultdict ，这是一种字典，当您引用尚未定义的键时，它会创建一个默认对象:

>>> from random import randint
>>> from collections import defaultdict
>>> 
>>> d = defaultdict(set)
>>> 
>>> for i in range(10):
...     el = randint(0, 5)
...     d[el].add(randint(0,10))
... 
>>> d
defaultdict(<type 'set'>, {0: set([2]), 1: set([7]), 2: set([8, 9, 5, 7]), 
3: set([1]), 4: set([1, 3, 4])})
>>> d[0]
set([2])
>>> d[2]
set([8, 9, 5, 7])