Сортировать элементы списка, принадлежащие перечисляемому объекту - PullRequest
Новая
       11

Сортировать элементы списка, принадлежащие перечисляемому объекту

0 голосов
/ 08 июня 2011

У меня есть список возможных видов продукции: 

PRODUCT_TYPE_CHOICES = [
    (PRODUCT_UNISEX, _(u"Unisex")),
    (PRODUCT_MENS, _(u"Mens")),
    (PRODUCT_WOMANS, _(u"Womans")),
]

И второй список, содержащий несколько объектов Product. 

all_products = [Product1, Product2, Product3, ..., ProductN]

У каждого продукта есть поле / свойство для выбора типа: 

type = models.CharField(_(u"Type"), max_length=15, choices=PRODUCT_TYPE_CHOICES, default=PRODUCT_UNISEX)

Теперь я хотел бы перечислить PRODUCT_TYPE_CHOICES и создать список для каждого перечисляемого элемента, который будет содержать его название и количество продуктов (из всех продуктов), имеющих тип перечислимого: 

ret = [['Unisex', 15],['Mens', 10],['Womans', 6]]

В настоящее время у меня есть это: 

for i, v in enumerate(PRODUCT_RODZAJ_CHOICES):
    r = []
    r.append(v[1])
    ret.append(r)

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

1 Ответ

0 голосов
/ 08 июня 2011

1 вкладыш: 

zip([sum(t) for t in zip( \
               *[(1,0,0) if p.type==PRODUCT_UNISEX \
                 else (0,1,0) if p.type==PRODUCT_MENS \
                 else (0,0,1) for p in products] \
                        ) \
    ], \
    ['Unisex','Mens','Womans'])

Или классическим способом: 

dict = {}
for p in products:
    dict[p.get_type_display()] = 1 + dict.get(p.get_type_display(), 0)

ret = [[key,value] for key,value in dict.iteritems()]
...