James McCool
Rename function create_stack_comparison to create_size_comparison in create_size_comparison.py for consistency with feature naming. This change aligns the function name with its purpose of comparing stack sizes.
add3343
| import pandas as pd | |
| def create_size_comparison(df: pd.DataFrame, entrants: list): | |
| overall_players = pd.Series(list(df['stack_size'])).value_counts() | |
| contest_len = len(df) | |
| set_frame = overall_players.to_frame().reset_index().rename(columns={'index': 'Player', 'count': 'Count'}) | |
| set_frame['Percent'] = set_frame['Count'] / contest_len | |
| set_frame = set_frame[['Player', 'Percent']] | |
| set_frame = set_frame.rename(columns={'Percent': f'Exposure Overall'}) | |
| player_frame = set_frame | |
| for each_user in entrants: | |
| overall_players = pd.Series(list(df[df['BaseName'] == each_user]['stack_size'])).value_counts() | |
| set_frame = overall_players.to_frame().reset_index().rename(columns={'index': 'Player', 'count': 'Count'}) | |
| set_frame['Percent'] = set_frame['Count'] / len(df[df['BaseName'] == each_user]) | |
| set_frame = set_frame[['Player', 'Percent']] | |
| set_frame = set_frame.rename(columns={'Percent': f'Exposure {each_user}'}) | |
| player_frame = pd.merge(player_frame, set_frame, on='Player', how='outer') | |
| return player_frame.sort_values(by='Exposure Overall', ascending=False) |