список Включить в список списков с использованием атрибута объекта

Question

У меня есть список объектов

objects = [<object 1>, <object 2>, <object 3>, <object 4>, <object 5>, <object 5>] 

Каждый объект имеет атрибут даты (object.date). Я хотел бы разбить список на список списков, в которых каждый подсписок имеет только объекты с одинаковой датой. Есть ли какой-нибудь умный способ сделать это? Моего наивное решение

list_of_lists = [] 
for object in objects: 
    success = 0 
    for list in list_of_lists: 
     if object.date == list[0].date: 
      list.append(object) 
      success = 1 
      break 
    if not success: 
     list_of_lists.append([object]) 

это действительно дает правильный результат, но я просто интересно об оптимальном решении.

Answer 1

Вы можете использовать itertools.groupby():

from itertools import groupby 
from operator import attrgetter 
my_lists = [list(g) for _, g in groupby(sorted(objects, key=attrgetter('date')), attrgetter('date'))] 

Обратите внимание, что если ваши объекты уже отсортированы по атрибуту даты, вам не нужно сортировать его.

Исходя из ваших потребностей, если вам нужно больше производительности с точкой зрения скорости выполнения вы можете использовать словарь для классификации объектов:

my_dict = {} 

for obj in objects: 
    my_dict.setdefault(obj.date, []).append(obj) # Don't forget to call the `date` if it's callable 

Обратите внимание, что если вы просто хотите списки в категоризированной форме вы можете получите значения словаря:

my_lists = my_dict.values()