52周存款挑战——可以任选几周的钱

#2020/10/21 第六节 52周存款挑战
#version:4 实现任取几周提取存款额
import math

def save_money_in_n_weeks(money_per_week, increase_money,money_want_list):

    money_list = []#记录每周存款数的列表

    for i in range (len(money_want_list)):
        #存钱
        money_list.append(money_per_week)
        saving=math.fsum(money_list)
        #输出消息
        print('第{}周，存入{}元，账户累计{}元'.format(money_want_list[i],money_per_week,saving))
        #更新下一周的金额
        money_per_week+=increase_money

def main():
    money_per_week=int(input('请输入每周存入的金额'))
    increase_money=float(input('请输入每周递增的金额'))
    total_week=int(input('请输入一共有多少周'))
    week_want1=int(input('请输入你想要的起始周'))
    week_want2=int(input('请输入你想要的末尾周'))
    money_want_list=list()
    for i in range (week_want1,week_want2+1):
        money_want_list.append(i)
    print('你想要的周有：',money_want_list)
    #调用函数
    save_money_in_n_weeks(money_per_week, increase_money,money_want_list)

if __name__=='__main__':
    main()