Update app.py

app.py

@@ -37,10 +37,16 @@ inv_dict = {"Month": t,
 inv_df = pd.DataFrame(inv_dict)
 inv_df.columns = inv_dict.keys()
 
+st.write(f"Table with returns at an interest rate of {r}")
 new_df = st.data_editor(inv_df)
 
-new_df["Returns"] = A * (1 + (new_df["Rate"] / (12*100)))**new_df["Month"]
-new_df["Continuous returns"] = A * np.exp(new_df["Rate"] * new_df["Month"] / (12*100))
+# New numpy array with the returns for each month in t
+new_y = A * (1 + (new_df["Rate"] / (12*100)))**new_df["Month"]
+new_df["Returns"] = new_y
+
+# New numpy array with the continuously compunded returns
+new_y_c = A * np.exp(new_df["Rate"] * new_df["Month"] / (12*100))
+new_df["Continuous returns"] = new_y_c
 
 st.dataframe(new_df)