lschlessinger commited on
Commit
7814537
·
1 Parent(s): 85f74f8

Update match_parser.py

Browse files
Files changed (1) hide show
  1. match_parser.py +10 -0
match_parser.py CHANGED
@@ -123,6 +123,16 @@ def get_num_competitions_played(df: pd.DataFrame, is_tournament: bool) -> int:
123
  return df[key_name].nunique()
124
 
125
 
 
 
 
 
 
 
 
 
 
 
126
  def get_current_rating(df: pd.DataFrame) -> int:
127
  return df.rating.iloc[0]
128
 
 
123
  return df[key_name].nunique()
124
 
125
 
126
+ def get_first_competition_year(df: pd.DataFrame, is_tournament: bool) -> int:
127
+ key_name = "tournament_end_date" if is_tournament else "event_date"
128
+ return df[key_name].min().year
129
+
130
+
131
+ def get_num_active_years(df: pd.DataFrame, is_tournament: bool) -> int:
132
+ key_name = "tournament_end_date" if is_tournament else "event_date"
133
+ return df[key_name].dt.year.nunique()
134
+
135
+
136
  def get_current_rating(df: pd.DataFrame) -> int:
137
  return df.rating.iloc[0]
138