Update app.py

app.py

@@ -1,4 +1,6 @@
 import streamlit as st
+import numpy as np
+import pandas as pd
 
 A = st.number_input(
     "Insert the initial investment (in $): "
@@ -7,10 +9,24 @@ r = st.number_input(
     "Insert the interest rate (nominal in %): "
 )
 T = st.number_input(
-    "Insert the number of years of your investment: ",
+    "Insert the number of months of your investment: ",
     step=1
 )
 
-y = A * (1 + (r/100))**T
+# Numpy array with the months
+t = np.arange(T + 1)
+
+# Numpy array with the returns for each month in t
+y = A * (1 + (r/(12*100)))**t
+
+# Create the investment dictionary
+inv_dict = {"Month": t, 
+           "Returns": y}
+
+# Create the investment dataframe
+inv_df = pd.DataFrame(inv_dict)
+inv_df.columns = inv_dict.keys()
+
+st.dataframe(inv_df)
 
 st.write(f"Your money after {T} years is ${y:.2f}")