如何使用python中的函数和循环，获得投资金额的每一年的复利输出？

def calculate_compound_interest(principal, interest_rate, years):
    amount = principal
    for i in range(years):
        amount += amount * interest_rate
        print("Year", i+1, ":", amount)
    return amount

principal = 1000  # 初始投资金额
interest_rate = 0.05  # 年利率
years = 5  # 投资年限

calculate_compound_interest(principal, interest_rate, years)