Update app.py

app.py

@@ -40,26 +40,20 @@ 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 numpy array with the returns for each month in t
-A_y = new_df["Returns"].shift(1)
 new_y = [A]
+new_y_c = [A]
 
-for i in range(1, len(A_y)):
+for i in range(1, len(t)):
     r_temp = new_df["Rate"].values[i]
+    # New numpy array with the returns for each month in t
     val =  new_y[i-1] * (1 + r_temp / (12*100))
     new_y.append(val)
 
-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]
-
-for i in range(1, len(A_c)):
-    r_temp = new_df["Rate"].values[i]
-    val = new_y_c[i-1] * np.exp(r_temp / (12*100))
-    new_y_c.append(val)
+    # New numpy array with the continuously compunded returns
+    val_c = new_y_c[i-1] * np.exp(r_temp / (12*100))
+    new_y_c.append(val_c)
 
+new_df["Returns"] = new_y
 new_df["Continuous returns"] = new_y_c
 
 st.dataframe(new_df)