2016-01-19 90 views
3

我對編程仍然很陌生,在這裏我遇到了一個問題,我似乎找不到一個好的答案。基本上,我做了一個小程序,可以讓你添加和刪除字典中的項目。這應該就像是一個來自RPG遊戲的庫存,您可以打開一個箱子,內容將被添加到庫存中,當您使用物品時,它們將從庫存中移除。我想設置它,如果清單中某個特定商品的數量爲零或更少,那麼該商品就會從字典中一起消失。這是我遇到麻煩的地方。這是我完成的程序如果值爲零,則從庫存(字典)中刪除項目

inventory = {} 

def add_to_inventory(item_tuple): 
    inventory[item_tuple[0]]=item_tuple[1] 


def add_all_items(tuple_list): 
    for tup in tuple_list: 
     if tup[0] not in inventory: 
      add_to_inventory(tup) 
     else: 
      inventory[tup[0]]+=tup[1] 
    return inventory 

def remove_items(item,quantity): 
    global inventory 
    updated_inventory={} 
    for key,value in inventory.items(): 
     if key==item: 
      value-=quantity 
      inventory[key]=value 
      if inventory[key]<= 0: 
       inventory[key]=0 
    for key,value in inventory.items(): 
     if value!=0: 
      updated_inventory[key]= value 
    inventory=updated_inventory 

這裏工作方案:

>>> chest=[("healing potion",3),("sword",1),("bread",5),("gold",50)] 
>>> inventory 
{} 
>>> add_all_items(chest) 
{'bread': 5, 'healing potion': 3, 'gold': 50, 'sword': 1} 
>>> remove_items("bread",3) 
>>> inventory 
{'bread': 2, 'healing potion': 3, 'gold': 50, 'sword': 1} 
>>> remove_items("bread",2) 
>>> inventory 
{'healing potion': 3, 'gold': 50, 'sword': 1} 
>>> 

我與前兩個功能是如何看起來很高興,但我remove_items功能看起來非常邋遢,我知道有即使它在技術上有效,也必須是更好的方式。我試着用「pop」和「del」,但是我會得到一個運行時錯誤,告訴我我的字典在迭代期間改變了大小。因此,現在我必須製作一個名爲updated_inventory的全新庫存,並且最終在迂迴過程之後將其設置爲等於「庫存」時纔是正確的。就像我剛纔說的那樣,我只有一個月的時間,所以我認爲在這裏我可以做得更好,但到目前爲止,這是唯一有效的方法。從閱讀中可以看出它看起來設計得很差。任何關於我能做得更好的建議?謝謝您的幫助。

+0

我想跟蹤清單中每個項目的數量。舉例來說,如果我有四種藥劑和兩種藥劑,我仍然希望庫存證明我有一些剩餘藥物。我只想要將該項目從字典中刪除,如果值<= 0。 –

回答

1

我不明白爲什麼需要搜索項目列表。字典的要點是可以立即訪問。

您也不需要複製字典。相反:

def remove_items(item,quantity): 
    global inventory 
    if item not in inventory: #make sure the item is in the inventory first 
     return 
    inventory[item] -= quantity 
    if inventory[item] <= 0: #remove item if the quantity is less than 0 
     inventory.pop(item,None) 
+0

我不敢相信我沒有想到這一點。我想我需要更多練習詞典。通過字典搜索沒有任何意義。我查看了你的答案,這對我來說更有意義。謝謝。 –

+1

爲什麼'彈出'項目與只是刪除項目 - '刪除庫存[項目]'? – AChampion

+0

既然任何工作,我的首選是彈出的情況下,我想對結果刪除的元素做一些事情。 – mattsap

相關問題