如何构建一个自定义函数？

Question

建立一个函数可根据商品原价与购买数量回传总购买价格，商品的折扣数如下：购买数量一件原价、两件九折、三件八折、四件以上五折，并利用该函数计算，当购买四项商品，其单位原价分别为100、110、120与130、以及购买数量分别为2、1、3与1件时的总购买价格。

Answer 1

def foo(number,price):     #函数接受两个参数，购买数量和产品原价
    if number == 1:  discount  = 1  #购买一件为原价不需要打折折扣不变。
    elif number == 2: discount = 0.9
    elif number == 3: discount = 0.8
    elif number >= 4: discount = 0.5
    return number*(price*discount) #计算产品打折后的总价格

if __name__ == "__main__":
    prices = [100, 110, 120, 130]
    numbers = [2, 1, 3, 1]
    total = 0
    for i in range(len(numbers)):
        tol = foo(numbers[i],prices[i])#计算出产品打折后总价格
        total = total + tol
    
    print("所有商品总价为{}".format(total))