Update app.py

app.py

@@ -41,12 +41,22 @@ st.write(f"Table with returns at an interest rate of {r}")
 new_df = st.data_editor(inv_df)
 
 # New numpy array with the returns for each month in t
-A = new_df["Returns"].shift(1)
-new_y = A * (1 + (new_df["Rate"] / (12*100)))
+A_y = new_df["Returns"].shift(1)
+
+new_y = [A]
+
+for i in range(1, len(A_y):
+    r_temp = new_df["Rate"].iloc(i)
+    val =  i * (1 + r_temp / (12*100))
+    new_y.append(val)
+
+
+new_y = A_y * (1 + (new_df["Rate"] / (12*100)))
 new_df["Returns"] = new_y
 
 # New numpy array with the continuously compunded returns
 A_c = new_df["Continuous returns"].shift(1)
+
 new_y_c = A_c * np.exp(new_df["Rate"] / (12*100))
 new_df["Continuous returns"] = new_y_c