aboutsummaryrefslogtreecommitdiff
path: root/src/ee/_utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/ee/_utils.py')
-rw-r--r--src/ee/_utils.py12
1 files changed, 12 insertions, 0 deletions
diff --git a/src/ee/_utils.py b/src/ee/_utils.py
new file mode 100644
index 0000000..598bca1
--- /dev/null
+++ b/src/ee/_utils.py
@@ -0,0 +1,12 @@
+import pandas as pd
+from typing import List
+
+def ensure_has_columns(df: pd.DataFrame, columns: List[str]):
+ all_columns = columns
+# print("all_columns={}".format(all_columns))
+# print("df={}".format(df.columns.tolist()))
+ for c in reversed(columns):
+ if not c in df.columns.tolist():
+ df.insert(0, column=c, value=pd.Series())
+# print("df={}".format(df.columns.tolist()))
+ return df