Adding new mean example for chris
This commit is contained in:
parent
739e5fb775
commit
8dbe5ecfdf
30
means_merge_test.py
Normal file
30
means_merge_test.py
Normal file
@ -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)
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user