futval_graph2.py

# futval_graph2.py

from graphics import *

def main():
    #Introduction
    print("This program plots the growth of a 10-year investment.")

    #Get principal and interest rate
    principal = float(input("Enter the initial principal: "))
    apr = float(input("Enter the annualized interest rate: "))

    #Create a graphics window with labels on the left edge
    win = GraphWin("Investment Growth Chart", 720, 500)
    win.setBackground("white")
    win.setCoords(-2, -200.0, 12, 10500.0)
    Text(Point(-1, 0), ' 0.0K').draw(win)
    Text(Point(-1, 2500), ' 2.5k').draw(win)
    Text(Point(-1, 5000), ' 5.0K').draw(win)
    Text(Point(-1, 7500), ' 7.5k').draw(win)
    Text(Point(-1, 10000), '10.0K').draw(win)

    #Draw bar for initial principal
    bar = Rectangle(Point(0,0), Point(1,principal))
    bar.setFill("green")
    bar.setWidth(2)
    bar.draw(win)

    #Draw bars for successive years
    for year in range(1, 11):
        #calculate value for the next year
        principal = principal*(1+apr)
        #draw bar for this value
        bar = Rectangle(Point(year, 0), Point(year+1, principal))
        bar.setFill("green")
        bar.setWidth(2)
        bar.draw(win)
        

    input("Press <Enter> to quit")
    win.close()

main()