python练习：函数写计算

用函数编写计算器，要求可以分别计算多个值的加/减/乘/除

def sum(args):    #定义加法函数
    count = 0
    for i in args:
        count += int(i)
    return count
def sub(args):    #定义减法函数
    count = int(args[0])    #起始值为第一个数
    for i in args[1:]:
        count -= int(i)
    return count
def mul(args):    #定义乘法函数
    count = 1    #起始值不能为0
    for i in args:
        count *= int(i)
    return count
def div(args):    #定义除法函数
    count = int(args[0])
    for i in args[1:]:
        count /= int(i)
    return count
a = input("请输入算式：")
if "+" in a:
    res = sum(a.split("+"))
elif "-" in a:
    res = sub(a.split("-"))
elif "*" in a:
    res = mul(a.split("*"))
elif "/" in a:
    res = div(a.split("/"))
print(res)