Adding new mean example for chris

This commit is contained in:
Tom Reincke 2026-04-20 23:17:21 +02:00
parent 739e5fb775
commit 8dbe5ecfdf

30
means_merge_test.py Normal file
View 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)