Python的学习笔记案例4--52周存钱挑战4.0

本节的内容主要为：灵活设置每周的存钱数，增加的存款数，存款周数以及将计算的过程封装在函数中，可以通过调用函数来计算累计存款金额。

"""
    作者：lanxingbudui
    功能：52周存钱挑战
    版本：v4.0
    日期：2019-03-07
    2.0新增功能：记录每周存钱数
    3.0新增功能：使用循环直接计数
    4.0新增功能：灵活设置每周的存钱数，增加的存款数及存款周数
"""
import math


def save_money_in_n_weeks(money_per_week, increase_money, total_week):
    """
        计算n周内的存款金额
    """

    money_list = []  # 累计存钱数的列表
    for i in range(total_week):
        # 计算累计存钱数
        money_list.append(money_per_week)
        saving = math.fsum(money_list)
        # 输出信息
        # print('第{}周，存入{}元，账户累计{}元'.format(i + 1, money_per_week, saving))
        # 循环递增
        money_per_week += increase_money

    return saving


def main():
    """
        主函数
    """
    money_per_week = float(input('请输入每周的存入的金额数：'))  # 每周存钱数
    increase_money = float(input('请输入每周的递增的金额数：'))   # 递增金额
    total_week = int(input('请输入总共的周数：'))      # 总的周数

    # 总的存款金额
    saving = save_money_in_n_weeks(money_per_week, increase_money, total_week)
    print('总的存款金额：', saving)


if __name__ == '__main__':
    main()