Preparing Game Data Starcraft 2 π π
df.to_parquet('sc2_actions.parquet', compression='snappy') If you control the game (bot development):
for player in replay.players: print(f"player.name (player.race) β MMR: player.mmr") Extract unit events, upgrades, resources, and positions: preparing game data starcraft 2
build_order_vector = [] for second in [60, 120, 180, 240, 300]: units_at_time = [e for e in replay.events if e.second <= second and e.name == 'UnitBornEvent'] build_order_vector.append(len([u for u in units_at_time if 'Zergling' in u.unit_type_name])) Goal: Predict race & opening from first 3 minutes. Extraction Code import sc2reader import pandas as pd replay = sc2reader.load_file("replay.SC2Replay") 'UpgradeCompleteEvent'] and event.second <
data = [] for event in replay.events: if event.name in ['UnitBornEvent', 'UpgradeCompleteEvent'] and event.second <= 180: data.append( 'time': event.second, 'type': event.name, 'unit': getattr(event, 'unit_type_name', None), 'upgrade': getattr(event, 'upgrade_type_name', None), 'player_race': event.player.play_race, 'winner': 1 if event.player == replay.winner else 0 ) = 180: data.append( 'time': event.second