mirror of
https://github.com/Steffo99/unimore-bda-3.git
synced 2024-11-21 23:44:21 +00:00
25 lines
891 B
Python
25 lines
891 B
Python
from unimore_bda_3.prelude import *
|
|
|
|
|
|
def add_scores(dataframe: pd.DataFrame):
|
|
for base in [1.01, 1.02, 1.03, 1.04, 1.05, 1.06, 1.07, 1.08, 1.09, 1.1, 1.11, 1.12, 1.13, 1.14, 1.15]:
|
|
score = pd.Series(dtype=float)
|
|
prev_date = None
|
|
for date, value in dataframe["Cumulative · Is something happening?"].items():
|
|
if not prev_date:
|
|
score[date] = 0
|
|
elif value:
|
|
score[date] = 1
|
|
else:
|
|
score[date] = score[prev_date] / base
|
|
prev_date = date
|
|
|
|
dataframe[f"Cumulative · General happening score with base {base}"] = score
|
|
dataframe[f"Cumulative · Scaled general happening score with base {base}"] = dataframe[f"Cumulative · General happening score with base {base}"] * dataframe["Google Trends · Score"]
|
|
|
|
return dataframe
|
|
|
|
|
|
__all__ = (
|
|
"add_scores",
|
|
)
|