Drop LSqft/Ac column if it exists

perfectly-preserved-pie · Jul 8, 2024 · 9f51e07 · 9f51e07
1 parent e5b3209
commit 9f51e07
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/buy_dataframe.py b/buy_dataframe.py
@@ -67,8 +67,9 @@
 # Merge all sheets into a single DataFrame
 df = pd.concat(xlsx.values())
 
-# Drop the LSqft/Ac column
-df = df.drop(columns=['LSqft/Ac'])
+# Drop the LSqft/Ac column only if it exists
+if 'LSqft/Ac' in df.columns:
+ df = df.drop(columns=['LSqft/Ac'])
 
 pd.set_option("display.precision", 10)