《Python编程快速上手——让繁琐工作自动化》实践项目——物品清单

小清高class2024-03-31 20:50

题目：

整个题目篇幅比较大。简化一下就是编写addToInventory函数并且返回一个字典。

思路：函数displayInventory用于物品清单的最后输出，而addToInventory是用于处理战利品的，在原来的装备上添加，已有的装备直接数量加1。而没有的添加到自己装备中，并且数量为1,之后再次出现数量直接加1。因此我们可以先用get方法来判断是否有该装备，也就是键，如果有，数量加1，没有就创建键值对{xx:1}。

python 复制代码

def addToInventory(inventory, addedItem):
    for i in addedItem:
        if inventory.get(i, 0) != 0:  # 如果不存在，get会返回备用值0
            inventory[i] += 1

        else:
            inventory.setdefault(i, 1)  # setdefault方法来创建新的键值对 
    return inventory

完整代码以及输出如下：

python 复制代码

def displayInventory(inventory):
    print("Inventory:")
    item_total = 0
    for k, v in inventory.items():
        print(str(v) + ' ' + k)
        item_total += v
    print("Total number of items: " + str(item_total))


def addToInventory(inventory, addedItem):
    for i in addedItem:
        if inventory.get(i, 0) != 0:
            inventory[i] += 1

        else:
            inventory.setdefault(i, 1)
    return inventory


if __name__ == '__main__':
    inv = {'gold coin': 42, 'rope': 1}
    dragonLoot = ['gold coin', 'dagger', 'gold coin', 'gold coin', 'ruby']
    inv = addToInventory(inv, dragonLoot)
    displayInventory(inv)

python 复制代码

Inventory:
45 gold coin
1 rope
1 dagger
1 ruby
Total number of items: 48