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

0 投票
1 回答
1104 浏览
提问于 2025-04-16 19:08

我有一个可能的产品类型列表:

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

撰写回答