itertools.groupby()分組字典列表

## itertools.groupby()分組字典列表數據

 

from operator import itemgetter
from itertools import groupby


students = [
    {'name': 'Peter', 'age': 19, 'score': 95},
    {'name': 'Lily', 'age': 22, 'score': 90},
    {'name': 'Stanley', 'age': 22, 'score': 92},
    {'name': 'Bob', 'age': 20, 'score': 88},
    {'name': 'Well', 'age': 20, 'score': 82}
]

for key, group in groupby(sorted(students, key=itemgetter("age")), key=itemgetter("age")):
    # groupby()函數同時返回分組關鍵字和一個與關鍵字相對應的可迭代對象
    # itemgetter()同樣可以接收多個關鍵字，也可以使用匿名函數代替此函數，但速度相比之下較慢
    print("Age: %s" % key)
    for g in group:
        print(g)
"""
Age: 19
{'name': 'Peter', 'age': 19, 'score': 95}
Age: 20
{'name': 'Bob', 'age': 20, 'score': 88}
{'name': 'Well', 'age': 20, 'score': 82}
Age: 22
{'name': 'Lily', 'age': 22, 'score': 90}
{'name': 'Stanley', 'age': 22, 'score': 92}

"""
# 也可以使用defaultdict()創建一個一對多字典方便進行隨機訪問
from collections import defaultdict


students_by_age = defaultdict(list)
for key, group in groupby(sorted(students, key=itemgetter("age")), key=itemgetter("age")):
    for i in group:
        students_by_age[key].append(i)

print(list(students_by_age[22]))
# [{'name': 'Lily', 'age': 22, 'score': 90}, {'name': 'Stanley', 'age': 22, 'score': 92}]

 

 

參考資料：
　　Python Cookbook, 3rd edition, by David Beazley and Brian K. Jones (O’Reilly).