From 1f1eb6b23b7d2c2697d8ef3f71c9d2e9e7c8897f Mon Sep 17 00:00:00 2001 From: Andrew Holt <15341747+ajh1143@users.noreply.github.com> Date: Thu, 18 Jan 2018 10:56:08 -0800 Subject: [PATCH] Create TidyDataShortcuts.py --- TidyDataShortcuts.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) create mode 100644 TidyDataShortcuts.py diff --git a/TidyDataShortcuts.py b/TidyDataShortcuts.py new file mode 100644 index 0000000..662f8e5 --- /dev/null +++ b/TidyDataShortcuts.py @@ -0,0 +1,17 @@ +import pandas as pd +import matplotlib.pyplot as plt + +#Melts a pandas DataFrame +def tidy_melt(df, exclude, var, value): + df_melt = pd.melt(df, id_vars= exclude, var_name=var, value_name=value) + return df_melt + +#Returns a pivot table from a tidy df +def tidy_to_pivot(df, indices, colname, valname): + df_piv = pd.pivot_table(df, index=indices, columns=colname, values=valname) + return df_piv + +#Returns a standard DataFrame from a tidydata set +def tidy_to_og(tidy_df): + restored_df = tidy_df.reset_index() + return restored_df