【python-列表/字典处理】使用groupby将列表元素分类

groupby.py

from itertools import groupby

def height_class(h):
    if h> 180:
        return "tall"
    elif h<160:
        return "short"
    else:
        return "middle"
    
friends = [191,144,142,170,177,188,293]
friends = sorted(friends, key = height_class)
for m, n in itertools.groupby(friends, key=height_class):
    print(m)
    print(list(n))
#middle
#[170, 177]
#short
#[144, 142]
#tall
#[191, 188, 293]