Как сгруппировать списки по двум элементам?

Question

[VaniLuksh]

Имеется список, состоящий из списков - ('Категория', 'Товар', 'Цена', 'Номер склада'). Надо сгруппировать списки по категории и номеру склада, так чтобы цены на каждый товар из категории на определенном складе тоже сложились.

list_in = [['Игрушки', 'Мяч надувной', '300', '1'], ['Игрушки', 'Бионикл', '8000', '1'], ['Ткань', 'Вельвет', '1000', '2'], ['Ткань', 'Джинса', '500', '2'], ['Игрушки', 'Бионикл', '1000', '2']]

Эти списки надо сгруппировать следующим образом:

list_out = [['Игрушки', '8300', '1'], ['Ткань', '1500', '2'], ['Игрушки', '1000', '2']]

Comments

[Vast Nectarine]

list_in = [['Игрушки', 'Мяч надувной', '300', '1'], ['Игрушки', 'Бионикл', '8000', '1'], ['Ткань', 'Вельвет', '1000', '2'], ['Ткань', 'Джинса', '500', '2'], ['Игрушки', 'Бионикл', '1000', '2']]
list_out = []
pos = {}
pos_counter = 0
for el in list_in:
    if (el[0], el[3]) not in pos.keys():
        pos[(el[0], el[3])] = pos_counter
        pos_counter += 1
        list_out.append([el[0],int(el[2]),el[3]])
    else:
        index = pos[(el[0], el[3])]
        list_out[index][1] += int(el[2])

print(list_out)

Сразу говорю, кто-то обязательно реализует эту задачу лучше меня, так что не стоит использовать моё решение.

UPD:

class Product:
    category: str
    product_name: str
    price: int
    warehouse_id: int
    def __init__(self, category: str, product_name: str, price: int, warehouse_id: int):
        self.category = category
        self.product_name = product_name
        self.price = price
        self.warehouse_id = warehouse_id
    
    @property
    def id(self):
        return (self.category, self.warehouse_id)
    

products = [
    Product('Игрушки', 'Мяч надувной', 300, 1),
    Product('Игрушки', 'Бионикл', 8000, 1),
    Product('Ткань', 'Вельвет', 1000, 2),
    Product('Ткань', 'Джинса', 500, 2),
    Product('Игрушки', 'Бионикл', 1000, 2)
]


result = []
product_ids = {}
product_counter = 0
for product in products:
    if product.id not in product_ids.keys():
        product_ids[product.id] = product_counter
        product_counter += 1
        result.append([product.category, product.price, product.warehouse_id])
    else:
        index = product_ids[product.id]
        result[index][1] += product.price

print(result)

[VaniLuksh]

Vast-Nectarine, спасибо!

Answer 1

[0xD34F]

key = lambda item: (item[0], item[3])
keys = set(map(key, list_in))
list_out = [ [ k[0], sum(int(n[2]) for n in list_in if k == key(n)), k[1] ] for k in keys ]

Comments

[VaniLuksh]

Спасибо! Подскажите, пожалуйста, как называется конструкция которую вы использовали в третьей строчке?

[0xD34F]

VaniLuksh, list comprehension.

Answer 2

[aRegius]

>>> from collections import defaultdict
>>> d = defaultdict(int)
>>> for i in list_in:
	        d[(i[0], i[-1])] += int(i[-2])

Comments

[VaniLuksh]

спасибо!

Answer 3

[Сергей П]

UPD: Поправил чуть способ со счетчиком:

In [1]: list_in = [['Игрушки', 'Мяч надувной', '300', '1'], ['Игрушки', 'Бионикл', '8000', '1'], ['Ткань', 'Вельвет', '1000', '2'], ['Ткань', 'Джинса', '500', '2'], ['Игрушки', 'Бионикл', '1000', '2']]

In [2]: from collections import Counter

In [3]: c = Counter(); list(map(c.update, ({(cat, stor): int(price)} for cat, name, price, stor in list_in)))
Out[3]: [None, None, None, None, None]

In [4]: c
Out[4]: Counter({('Игрушки', '1'): 8300, ('Игрушки', '2'): 1000, ('Ткань', '2'): 1500})

In [5]: [[*k, v] for k, v in c.items()]
Out[5]: [['Игрушки', '1', 8300], ['Ткань', '2', 1500], ['Игрушки', '2', 1000]]

Но иногда нужно не просто считать, а что-то еще более сложное делать после группировки, тогда больше подойдёт вот такой вариант:

In [11]: from itertools import groupby                                                              

In [12]: [[*k, sum(int(rr[2]) for rr in v)] for k, v in groupby(list_in, lambda r: (r[0], r[-1]))]
Out[12]: [['Игрушки', '1', 8300], ['Ткань', '2', 1500], ['Игрушки', '2', 1000]]

Тут синтаксис третьего питона. Для второго придётся чуть поменять со звёздочкой.