Python计算合理避税后收入增加情况

假设张三所在团队有一小笔收入，但是超出800元以上的部分要交20%的税，为了提高总收入，可以申报多人合作完成，这样的话每人800之内的部分都不需要扣税，那么总收入能提高多少呢？

>>> def calcFee(total, num):

    '''total为原始总收入，num为申报人数'''

    # 只申报一人的收入

    origin = 800 + (total-800)*0.8

    # 申报多人完成时不扣税的基数

    base = num*800

    # 人均收入小于等于800的情况

    if total <= base:

        return (total, total-origin)

    now = base + (total-base)*0.8

    # 返回结果(实际收入，提高的收入)

    return (now, now-origin)

>>> calcFee(3000, 10)

(3000, 440.0)

>>> calcFee(3000, 5)

(3000, 440.0)

>>> calcFee(5000, 5)

(4800.0, 640.0)

>>> calcFee(45000, 5)

(36800.0, 640.0)

>>> calcFee(45000, 15)

(38400.0, 2240.0)

>>> calcFee(450000, 85)

(373600.0, 13440.0)