Translate team names

daily_weekly_leaderboard.py

@@ -18,6 +18,7 @@ df = (
     .rename({
         'name': 'Name',
         'release_speed': 'Velocity',
+        'team': 'Team'
     })
 )
 

data.py

@@ -12,6 +12,7 @@ from seasons import SEASONS
 from translate import (
     translate_pa_outcome, translate_pitch_outcome,
     jp_pitch_to_en_pitch, jp_pitch_to_pitch_code,
+    jp_team_to_en_team, jp_team_to_en_full_team,
     max_pitch_types
 )
 
@@ -66,6 +67,21 @@ for season in SEASONS:
     # load player data
     _player_df = pl.read_csv(os.path.join(season_dir, 'player.csv'))
 
+    # translate game data
+    _game_df = (
+        _game_df
+        .with_columns(
+            pl.col('home_team').alias('jp_home_team'),
+            pl.col('away_team').alias('jp_away_team')
+        )
+        .with_columns(
+            pl.col('home_team').replace_strict(jp_team_to_en_team),
+            pl.col('home_team').replace_strict(jp_team_to_en_full_team).alias('full_home_team'),
+            pl.col('away_team').replace_strict(jp_team_to_en_team),
+            pl.col('away_team').replace_strict(jp_team_to_en_full_team).alias('full_away_team')
+        )
+    )
+
     # translate pa data
     _pa_df = (
         _pa_df
@@ -146,7 +162,6 @@ for season in SEASONS:
         pl.read_csv(os.path.join(season_dir, 'register.csv'))
         .with_columns(
             pl.col('en_name').str.replace(',', '').alias('en_name'),
-
         )
         .select(
             pl.col('en_name'),
@@ -154,7 +169,19 @@ for season in SEASONS:
             pl.col('jp_name').alias('name')
         )
     )
-    _player_df = _player_df.join(register, on=['name', 'team'], how='inner').with_columns(pl.col('en_name').alias('name')).drop(pl.col('en_name'))
+    _player_df = (
+        _player_df
+        .join(register, on=['name', 'team'], how='inner')
+        .with_columns(
+            pl.col('en_name').alias('name'),
+            pl.col('team').alias('jp_team')
+        )
+        .with_columns(
+            pl.col('jp_team').replace_strict(jp_team_to_en_team).alias('team'),
+            pl.col('jp_team').replace_strict(jp_team_to_en_full_team).alias('full_team'),
+        )
+        .drop(pl.col('en_name'))
+    )
 
     # merge pitch and pa data
     _df = (
@@ -261,3 +288,8 @@ def compute_league_pitch_stats(df):
   return df.group_by('pitch_name').agg(pl.col('release_speed').mean().round(1).alias('Velocity'))
 
 league_pitch_stats = compute_league_pitch_stats(df)
+
+if __name__ == '__main__':
+    print(df.shape)
+    print(df.columns)
+    breakpoint()

translate.py

@@ -204,3 +204,57 @@ def translate_pitch_outcome(outcome):
   return outcome
 
 max_pitch_types = len(jp_pitch_to_en_pitch)
+
+jp_team_to_en_team = {
+    '阪神': 'Hanshin',
+    '楽天': 'Rakuten',
+    'オリックス': 'Orix',
+    'ソフトバンク': 'SoftBank',
+    'ヤクルト': 'Yakult',
+    'ロッテ': 'Lotte',
+    '巨人': 'Giants',
+    '日本ハム': 'Nippon-Ham',
+    '西武': 'Seibu',
+    '中日': 'Chunichi',
+    'DeNA': 'DeNA',
+    '広島': 'Hiroshima',
+    'オリックス・バファローズ': 'Orix',
+    '北海道日本ハムファイターズ': 'Nippon-Ham',
+    '埼玉西武ライオンズ': 'Seibu',
+    '東京ヤクルトスワローズ': 'Yakult',
+    '横浜DeNAベイスターズ': 'DeNA',
+    '読売ジャイアンツ': 'Giants',
+    '千葉ロッテマリーンズ': 'Lotte',
+    '広島東洋カープ': 'Hiroshima',
+    '東北楽天ゴールデンイーグルス': 'Rakuten',
+    '中日ドラゴンズ': 'Chunichi',
+    '福岡ソフトバンクホークス': 'SoftBank',
+    '阪神タイガース': 'Hanshin'
+}
+
+jp_team_to_en_full_team = {
+    '阪神': 'Hanshin Tigers',
+    '楽天': 'Rakuten Eagles',
+    'オリックス': 'Orix Buffaloes',
+    'ソフトバンク': 'Softbank Hawks',
+    'ヤクルト': 'Yakult Swallows',
+    'ロッテ': 'Lotte Marines',
+    '巨人': 'Tokyo Giants',
+    '日本ハム': 'Nippon-Ham Fighters',
+    '西武': 'Seibu Lions',
+    '中日': 'Chunichi Dragons',
+    'DeNA': 'DeNA BayStars',
+    '広島': 'Hiroshima Carp',
+    'オリックス・バファローズ': 'Orix Buffaloes',
+    '北海道日本ハムファイターズ': 'Nippon-Ham Fighters',
+    '埼玉西武ライオンズ': 'Seibu Lions',
+    '東京ヤクルトスワローズ': 'Yakult Swallows',
+    '横浜DeNAベイスターズ': 'DeNA BayStars',
+    '読売ジャイアンツ': 'Yomiuri Giants',
+    '千葉ロッテマリーンズ': 'Lotte Marines',
+    '広島東洋カープ': 'Hiroshima Caro',
+    '東北楽天ゴールデンイーグルス': 'Rakuten Eagles',
+    '中日ドラゴンズ': 'Chunichi Dragons',
+    '福岡ソフトバンクホークス': 'SoftBank Hawks',
+    '阪神タイガース': 'Hanshin Tigers'
+}