diff options
Diffstat (limited to 'src/ee/_utils.py')
-rw-r--r-- | src/ee/_utils.py | 12 |
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 |