diff --git a/means_merge_test.py b/means_merge_test.py new file mode 100644 index 0000000..396c2e0 --- /dev/null +++ b/means_merge_test.py @@ -0,0 +1,30 @@ +#!/usr/bin/python3 +import pandas as pd +import sys + +def imp_selector(index): + return "attr1" if i % 2 == 0 else "attr2" + +a = [] +imp_attr_list = {} +for i in range(50): + a.append((i,i+10)) + imp_attr_list[i]=imp_selector(i) + +df = pd.DataFrame.from_records(a, columns=["attr1", "attr2"]) + +m_list=[] +mean_list= dict() +for v in set(imp_attr_list.values()): + rolling_mean = df[v].rolling(7).mean() + mean_list[v] = rolling_mean.values + +# for k,v in mean_list.items(): + # print(k,v) +for i in range(df.shape[0]): + value = mean_list[imp_attr_list[i]][i] + df.loc[i, "7_days_means"] = value + +print(df) + +