Skip to content

Instantly share code, notes, and snippets.

@databyjp

databyjp/sixdegs_leagues_connection_locs.py

Created March 14, 2020 05:46
Show Gist options
  • Save databyjp/8413e8fe1b1b34be7e4dd73ae3c1bc64 to your computer and use it in GitHub Desktop.
Save databyjp/8413e8fe1b1b34be7e4dd73ae3c1bc64 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
sixdegs_leagues_connection_locs.py
teams_contacted_df = teams_contacted_df.assign(orig_lat=0, orig_lon=0, dest_lat=0, dest_lon=0)
for temp_tuple in teams_contacted_df.itertuples():
if temp_tuple.con_from == None:
from_name = temp_tuple.team
else:
from_name = temp_tuple.con_from
teams_contacted_df.loc[temp_tuple.Index, 'orig_lat'] = arena_df[arena_df.teamupper == from_name]['lat'].values[0]
teams_contacted_df.loc[temp_tuple.Index, 'orig_lon'] = arena_df[arena_df.teamupper == from_name]['lon'].values[0]
teams_contacted_df.loc[temp_tuple.Index, 'dest_lat'] = arena_df[arena_df.teamupper == temp_tuple.team]['lat'].values[0]
teams_contacted_df.loc[temp_tuple.Index, 'dest_lon'] = arena_df[arena_df.teamupper == temp_tuple.team]['lon'].values[0]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment