Какой идиоматический Python эквивалентен шаблонному тегу Django 'regroup'?

Question

http://docs.djangoproject.com/en/dev/ref/templates/builtins/#regroup

Я могу придумать несколько способов сделать это с помощью циклов, но мне бы особенно хотелось узнать, есть ли аккуратная однострочная.

Answer 1

Объедините itertools.groupby с operator.itemgetter, чтобы получить довольно хорошее решение:

from operator import itemgetter
from itertools import groupby

key = itemgetter('gender')
iter = groupby(sorted(people, key=key), key=key)

for gender, people in iter:
    print '===', gender, '==='
    for person in people:
        print person

Answer 2

Предыдущие ответы помогли мне решить мою проблему. Для дальнейшего использования если у вас есть некоторые вложенные данные, такие как

{'city_name': 'City1', 'comp_name': 'Company1', 'name': 'Branch1'}

и вы хотите сгруппировать по городам, а затем по компаниям в этом городе, например:

City1
 Company 1
   Branch 1
   Branch 2
 Company 2
   Branch 1
 Company 3
   Branch 1
City2
 Company 2
   Branch 1
 Company 3
   Branch 1
   Branch 2
City3
 Company 1
   Branch 1
 Company 2
   Branch 1
   Branch 2

Я решил это следующим образом:

key = itemgetter('city_name')    
iter = groupby(queryset, key=key) # assuming queryset is already sorted by city_name

for key, group in iter:
    print(key)
    key2 = itemgetter('company_name')
    iter2 = groupby(sorted(group, key=key2), key=key2) # now we must sort by company_name
    for comp, branch in iter2:
        print(comp)
        for b in branch:
            print(b)

Answer 3

Если источник данных (people в данном случае) уже отсортирован по ключу, вы можете пропустить sorted вызов:

iter = groupby(people, key=lambda x:x['gender'])
for gender, people in iter:
    print '===', gender, '==='
    for person in people:
        print person

Примечание: если sorted является общим словарем, нет гарантий порядка; поэтому вы должны позвонить sorted. Здесь я предполагаю, что sorted - это collections.OrderedDict или какая-то другая упорядоченная структура данных.