collections模块下的defaultdict用法

defaultdict

from collections import defaultdict
s=[('yellow',1),('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
d=defaultdict(list)  # 当键不存在时，添加键并给出默认值
for k, v in s:
    d[k].append(v)
    print(d)
a=sorted(d.items())
print(a)

"""
defaultdict(<class 'list'>, {'yellow': [1]})
defaultdict(<class 'list'>, {'yellow': [1], 'blue': [2]})
defaultdict(<class 'list'>, {'yellow': [1, 3], 'blue': [2]})
defaultdict(<class 'list'>, {'yellow': [1, 3], 'blue': [2, 4]})
defaultdict(<class 'list'>, {'yellow': [1, 3], 'blue': [2, 4], 'red': [1]})
[('blue', [2, 4]), ('red', [1]), ('yellow', [1, 3])]
"""

print("-----------------------------------------------------")
也可以用setdefault改写
s = [('yellow', 1), ('blue', 2), ('yellow', 3), ('blue', 4), ('red', 1)]
d={}
for k, v in s:
    d.setdefault(k,[]).append(v)  # # 当键不存在时，添加键并给出默认值
print('\n',d)
a=sorted(d.items())
print('\n',a)