cyberosa commited on
Commit
4e4eb03
·
1 Parent(s): 9c51f62

fixing date format

Browse files
Files changed (1) hide show
  1. 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