cyberosa
commited on
Commit
·
4e4eb03
1
Parent(s):
9c51f62
fixing date format
Browse files- tabs/agent_graphs.py +7 -0
tabs/agent_graphs.py
CHANGED
@@ -87,6 +87,9 @@ def plot_rolling_average_roi(
|
|
87 |
traders_data["trader_address"].isin(unique_addresses)
|
88 |
]
|
89 |
# create the date column
|
|
|
|
|
|
|
90 |
filtered_traders_data["creation_date"] = filtered_traders_data[
|
91 |
"creation_timestamp"
|
92 |
].dt.date
|
@@ -117,12 +120,16 @@ def get_twoweeks_rolling_average_roi(traders_data: pd.DataFrame) -> pd.DataFrame
|
|
117 |
|
118 |
# Aggregate ROI at the date level
|
119 |
daily_avg = traders_data.groupby("creation_date")["roi"].mean().reset_index()
|
|
|
|
|
120 |
daily_avg = daily_avg.set_index("creation_date")
|
121 |
# Now resample and rolling
|
122 |
weekly_avg = daily_avg.resample("W").mean()
|
123 |
rolling_avg = weekly_avg.rolling(window=2).mean().reset_index()
|
124 |
|
125 |
rolling_avg.rename(columns={"roi": "rolling_avg_roi"}, inplace=True)
|
|
|
|
|
126 |
return rolling_avg
|
127 |
|
128 |
|
|
|
87 |
traders_data["trader_address"].isin(unique_addresses)
|
88 |
]
|
89 |
# create the date column
|
90 |
+
filtered_traders_data["creation_timestamp"] = pd.to_datetime(
|
91 |
+
filtered_traders_data["creation_timestamp"], timezone="UTC"
|
92 |
+
)
|
93 |
filtered_traders_data["creation_date"] = filtered_traders_data[
|
94 |
"creation_timestamp"
|
95 |
].dt.date
|
|
|
120 |
|
121 |
# Aggregate ROI at the date level
|
122 |
daily_avg = traders_data.groupby("creation_date")["roi"].mean().reset_index()
|
123 |
+
# Ensure creation_date is datetime64[ns]
|
124 |
+
daily_avg["creation_date"] = pd.to_datetime(daily_avg["creation_date"])
|
125 |
daily_avg = daily_avg.set_index("creation_date")
|
126 |
# Now resample and rolling
|
127 |
weekly_avg = daily_avg.resample("W").mean()
|
128 |
rolling_avg = weekly_avg.rolling(window=2).mean().reset_index()
|
129 |
|
130 |
rolling_avg.rename(columns={"roi": "rolling_avg_roi"}, inplace=True)
|
131 |
+
# Optionally, rename the date column for clarity
|
132 |
+
rolling_avg.rename(columns={"creation_date": "month_year_week"}, inplace=True)
|
133 |
return rolling_avg
|
134 |
|
135 |
|