对可枚举对象的列表元素进行排序
我有一个可能的产品类型列表:
PRODUCT_TYPE_CHOICES = [
(PRODUCT_UNISEX, _(u"Unisex")),
(PRODUCT_MENS, _(u"Mens")),
(PRODUCT_WOMANS, _(u"Womans")),
]
还有一个包含一些产品对象的第二个列表。
all_products = [Product1, Product2, Product3, ..., ProductN]
每个产品都有一个字段/属性来选择类型:
type = models.CharField(_(u"Type"), max_length=15, choices=PRODUCT_TYPE_CHOICES, default=PRODUCT_UNISEX)
现在我想列出所有的产品类型选择,并为每种类型创建一个列表,里面包含它的名称和所有产品中属于该类型的产品数量:
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
一行代码:
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()]