TR 2 update

Author: iansedano

pandas-iterate-over-rows/cumulative_sum_codetiming.py

@@ -5,12 +5,11 @@
 def loop_cumsum(products):
     cumulative_sum = []
     for product in products.itertuples():
+        income = product.sales * product.unit_price
         if cumulative_sum:
-            cumulative_sum.append(
-                cumulative_sum[-1] + (product.sales * product.unit_price)
-            )
+            cumulative_sum.append(cumulative_sum[-1] + income)
         else:
-            cumulative_sum.append(product.sales * product.unit_price)
+            cumulative_sum.append(income)
     return products.assign(cumulative_income=cumulative_sum)
 
 

pandas-iterate-over-rows/cumulative_sum_perfplot.py

@@ -5,12 +5,11 @@
 def loop_cumsum(products):
     cumulative_sum = []
     for product in products.itertuples():
+        income = product.sales * product.unit_price
         if cumulative_sum:
-            cumulative_sum.append(
-                cumulative_sum[-1] + (product.sales * product.unit_price)
-            )
+            cumulative_sum.append(cumulative_sum[-1] + income)
         else:
-            cumulative_sum.append(product.sales * product.unit_price)
+            cumulative_sum.append(income)
     return products.assign(cumulative_income=cumulative_sum)
 
 

pandas-iterate-over-rows/products.py

@@ -1,7 +1,20 @@
 # %%
-
 import pandas as pd
 
+# %% Get the cumulative sum with .itertuples()
+products = pd.read_csv("resources/products.csv")
+
+cumulative_sum = []
+
+for product in products.itertuples():
+    income = product.sales * product.unit_price
+    if cumulative_sum:
+        cumulative_sum.append(cumulative_sum[-1] + income)
+    else:
+        cumulative_sum.append(income)
+
+products.assign(cumulative_income=cumulative_sum)
+
 # %% To get cumulative sum, instead of looping, you can create intermediate
 # columns and use .cumsum()
 products = (
@@ -12,19 +25,3 @@
     )
     .drop(columns="income")
 )
-
-# %% The equivalent way to do that with only .itertuples()
-products = pd.read_csv("resources/products.csv")
-
-cumulative_sum = []
-
-for product in products.itertuples():
-    if cumulative_sum:
-        cumulative_sum.append(
-            cumulative_sum[-1] + (product.sales * product.unit_price)
-        )
-    else:
-        cumulative_sum.append(product.sales * product.unit_price)
-
-products.assign(cumulative_income=cumulative_sum)
-# %%