对属于可枚举对象的列表元素进行排序

时间:2011-06-07 22:02:57

标签: python django enumerable

我列出了可能的产品类型:

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)

所以我缺少每种类型的产品数量。我试图用lambda做这个,但目前没有任何成功。

1 个答案:

答案 0 :(得分:0)

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()]