Update solution to invest challenge

somacdivad · somacdivad · commit d98ce064a9a1 · 2019-03-10T04:53:48.000-05:00
diff --git a/ch06-functions-and-loops/5-challenge.py b/ch06-functions-and-loops/5-challenge.py
@@ -4,15 +4,14 @@
 
 # Calculate compound interest to track the growth of an investment
 
-
-def invest(amount, rate, time):
-    print(f"principal amount: ${amount}")
-    print("annual rate of return:", rate)
-    for t in range(1, time + 1):
+def invest(amount, rate, years):
+    for year in range(1, years + 1):
         amount = amount * (1 + rate)
-        print(f"year {t}: ${amount:,.2f}")
-    print()
+        print(f"year {year}: ${amount:,.2f}")
+
 
+amount = float(input("Enter a principal amount: "))
+rate = float(input("Enter an anual rate of return: "))
+years = int(input("Enter a number of years: "))
 
-invest(100, 0.05, 8)
-invest(2000, 0.025, 5)
+invest(amount, rate, years)
