示例中的打折方式怎么完成？

Question

消费2000，500以内不打折，500-1100的金额是600，打8折；1100-1600的金额是500，打7折；1600以上金额是400打6折，总付款为500+6000.8+5000.7+400*0.6=1570？

Answer 1

def discount(n):
    if n < 500: return n
    if n < 1100: return (n-500)*0.8+500
    if n < 1600: return (n-1100)*0.7+600*0.8+500
    return (n-1600)*0.6+500*0.7+600*0.8+500

print(discount(2000))