计算等额本息还款计划详情数据的python代码

在Python中，计算等额本息还款计划通常涉及到一些数学公式，例如本金、利息和每月还款额的计算。这里是一个简单的例子，假设年利率（r）、贷款期限（n）和首期付款（pmt0）已知：

def calculate_amortization(r, n, pmt0):
    # 将年利率转换为月利率
    r_month = r / 12
    
    # 等额本息公式：每月偿还额 = [贷款本金 x 月利率 x (1 + 月利率)^(贷款期数)] / [(1 + 月利率) ^ 贷款期数 - 1]
    pmnt = pmt0 if pmt0 else (r_month * n * (1 + r_month)**n) / ((1 + r_month)**n - 1)
    
    # 总利息和总还款额
    total_interest = pmtn * n - pmt0 * (1 - (1 + r_month)**(-n))
    total_repayment = pmtn * n
    
    return {
        "每月还款额": round(pmnt, 2),  # 还款金额保留两位小数
        "总利息": round(total_interest, 2),
        "总还款额": round(total_repayment, 2),
    }

# 示例用法
r = 0.05  # 年利率，比如5%
n = 360  # 贷款期限，例如30年
pmt0 = 10000  # 首期付款

amortization_data = calculate_amortization(r, n, pmt0)
print(amortization_data)