题目:
整个题目篇幅比较大。简化一下就是编写addToInventory函数并且返回一个字典。
思路:函数displayInventory用于物品清单的最后输出,而addToInventory是用于处理战利品的,在原来的装备上添加,已有的装备直接数量加1。而没有的添加到自己装备中,并且数量为1,之后再次出现数量直接加1。因此我们可以先用get方法来判断是否有该装备,也就是键,如果有,数量加1,没有就创建键值对{xx:1}。
def addToInventory(inventory, addedItem):for i in addedItem:if inventory.get(i, 0) != 0: # 如果不存在,get会返回备用值0inventory[i] += 1else:inventory.setdefault(i, 1) # setdefault方法来创建新的键值对 return inventory
完整代码以及输出如下:
def displayInventory(inventory):print("Inventory:")item_total = 0for k, v in inventory.items():print(str(v) + ' ' + k)item_total += vprint("Total number of items: " + str(item_total))def addToInventory(inventory, addedItem):for i in addedItem:if inventory.get(i, 0) != 0:inventory[i] += 1else:inventory.setdefault(i, 1)return inventoryif __name__ == '__main__':inv = {'gold coin': 42, 'rope': 1}dragonLoot = ['gold coin', 'dagger', 'gold coin', 'gold coin', 'ruby']inv = addToInventory(inv, dragonLoot)displayInventory(inv)
Inventory:
45 gold coin
1 rope
1 dagger
1 ruby
Total number of items: 48