以按月产生的罚息为基期计算复利的python代码

在Python中，我们可以使用内置的`math`模块来进行复利计算，假设每月的罚息是固定的，我们首先需要计算出总的利息，然后加上本金。以下是一个简单的示例，假设初始本金、每月罚息率、月份数以及是否从第一个月开始计息：

import math

# 定义函数计算复利
def compound_interest(principal, monthly_interest_rate, months, start_from_first_month=True):
    # 确保利率在0到1之间
    if monthly_interest_rate > 1 or monthly_interest_rate < 0:
        raise ValueError("每月罚息率必须在0到1之间")

    # 如果是从第一个月开始计息，那么第一个月的利息是本金乘以利率
    if start_from_first_month:
        interest_first_month = principal * monthly_interest_rate
    else:
        interest_first_month = 0

    # 计算总利息
    total_interest = interest_first_month
    for _ in range(1, months):
        interest_current_month = (principal + total_interest) * monthly_interest_rate
        total_interest += interest_current_month

    # 返回最终金额（本金加总利息）
    return principal + total_interest

# 示例使用
initial_amount = 1000  # 初始本金
monthly_penalty_rate = 0.05 / 12  # 每月罚息率（例如5%年化分摊到每个月）
months = 12  # 总月数
result = compound_interest(initial_amount, monthly_penalty_rate, months)

print(f"经过{months}个月后的总额是：{result}")