исправьте меня для использования генераторов или скажите мне другим способом

Question

У меня есть пункт меню в качестве ключа и цена как значение. Там может существовать сочетание предметов, которые будут немного дешевле, чем отдельные предметы. Для EXA:

menu = { 
    ('burger',) : 5.00, 
    ('pizza',) : 12.00, 
    ('coke',) : 4.00, 
    ('macpuff',) : 4.00, 
    ('pasta',) : 3.00, 
    ('french_fries',) : 2.00, 
    ('burger', 'coke', 'french_fries') : 10.00, 
    ('pizza', 'coke') : 15.00, 
} 

Теперь предположим, что я заказал несколько пунктов, то выход будет мин сумма данного заказа:

I/P > burger, coke 
O/P > 9  (5.00 + 4.00) 

I/P > burger, coke, french_fries 
O/P > 10.00 

I/P > pizza, coke, french_fries 
O/P > 17.00 (15.00 + 2.00) 

здесь код я попытался так для всех цен, которые я буду использовать в качестве генераторов :

def isSubset(a, b): 
    """ 
     compare two iterable and return true if first is subset of second 
    """ 
    b = list(b) 
    if not hasattr(a, '__iter__'): 
     a = [a] 
    for each in a: 
     try: 
      b.remove(each) 
     except ValueError: 
      return False 
    return True 

def rest_min_price(order): 
    if order: 
     for item, price in menu.iteritems(): 
      if isSubset(order[0], item): 
       new_order = order[1:] 
       for itm in item: 
        try: 
         new_order.remove(itm) 
        except ValueError: 
         pass 
       yield price + rest_min_price(new_order) 

, но когда я запускаю это он говорит об ошибке типа:

for each in rest_min_price(order_item): 
    print each 

TypeError: unsupported operand type(s) for +: 'int' and 'generator' 

Answer 1

Спасибо всем за ваш ответ. где-то я использовал ваше предложение, и теперь я решил его по тому же подходу с другой реализацией.

вот мой код:

class Rest_Menu(): 
    def __init__(self, menu): 
     self.menu = menu 
     self.all_price = [] 

    def min_price(self, order, total_price=0): 
     """ 
     Return minm menu price by calculating all possible combination. 
     """ 
     if order: 
      for item, price in self.menu.iteritems(): 
       if isSubset(order[0], item): 
        new_order = [each for each in order] 
        for itm in item: 
         try: 
          new_order.remove(itm) 
         except ValueError: 
          pass 
        self.min_price(new_order, price+total_price) 
     else: 
      self.all_price.append(total_price) 
     return min(self.all_price) 

Еще раз спасибо. :)

Answer 2

Ваша проблема возникает в вашей попытке объединить генератор с рекурсией:

yield price + rest_min_price(new_order) 

здесь price является int, но rest_min_price() является generator (потому что вам yield, а не return, его результат), поэтому вы не можете добавьте их вместе и получите TypeError.

Вы должны перебрать элементы в вашем генераторе:

for item in rest_min_price(order_item): 
    # process the item 
    # yield result with price 

Answer 3

Вы поняли yield. Функция, которая содержит yield, волшебным образом становится генератором. Если вы просто назовете это, вы получите объект-генератор, который вы не можете использовать напрямую; вам нужно вызвать его метод next(), чтобы вынуть из него значения. Чаще всего вы помещаете его в цикл for.

>>> def genDemo(n): 
... while n > 0: 
...  yield n 
...  n -= 1 
... 
>>> 
>>> gd = genDemo(3) 
>>> gd 
<generator object genDemo at 0x7fce12152820> 
>>> gd.next() 
3 
>>> gd.next() 
2 
>>> gd.next() 
1 
>>> gd.next() 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
StopIteration 
>>> for x in genDemo(3): 
... print x 
... 
3 
2 
1 
>>> _ 

Ваш код является просто нормальной рекурсивной функцией; он может использовать return.

Генераторы удобны, когда вы не хотите получать всю последовательность в памяти; это имеет смысл с очень длинными и бесконечными последовательностями. Вы можете генерировать перестановки до тех пор, пока не найдете нужную вам информацию или не создадите случайные числа, пока не найдете нужную вам информацию, или не прочитайте ее из сокета, пока не получите сообщение о выходе, и т. Д.

Что касается подмножеств: вы не можете проверить для подмножеств лениво, если ваши данные не упорядочены одинаково. С наборами данных у вас будет лучше использовать встроенный set.