block1.py

import random
import numpy as np
import matplotlib.pyplot as plt
import scipy.stats as stats
import seaborn as sns
import math

def EstimateReturns(initial_invest, expected_returns, std_dev, duration):
    final_money = []
    yearcount = []
    
    year = 1
    
    for year in range(duration):
        this_return = np.random.normal(expected_returns,std_dev)
        initial_invest = initial_invest * (1+this_return)
        final_money.append(initial_invest)
        yearcount.append(year)
        
    plt.ylabel('Amount of $')
    plt.xlabel('Number of years')
    plt.plot(yearcount,final_money)
    
    final_funds.append(final_money[-1])
    return(final_funds)