代码讲解 def matches(g_matches): g_matches.insert(2, 'potential1', g_matches['country1'].map(squad_stats.set_index('nationality_name')['potential'])) g_matches.insert(3, 'potential2', g_matches['country2'].map(squad_stats.set_index('nationality_name')['potential'])) g_matches.insert(4, 'rank1', g_matches['country1'].map(last_team_scores.set_index('team')['rank'])) g_matches.insert(5, 'rank2', g_matches['country2'].map(last_team_scores.set_index('team')['rank'])) pred_set = [] for index, row in g_matches.iterrows(): if row['potential1'] > row['potential2'] and abs(row['potential1'] - row['potential2']) > 2: pred_set.append({'Team1': row['country1'], 'Team2': row['country2']}) elif row['potential2'] > row['potential1'] and abs(row['potential2'] - row['potential1']) > 2: pred_set.append({'Team1': row['country2'], 'Team2': row['country1']}) else: if row['rank1'] > row['rank2']: pred_set.append({'Team1': row['country1'], 'Team2': row['country2']}) else: pred_set.append({'Team1': row['country2'], 'Team2': row['country1']}) pred_set = pd.DataFrame(pred_set) pred_set.insert(2, 'Team1_FIFA_RANK', pred_set['Team1'].map(last_team_scores.set_index('team')['rank'])) pred_set.insert(3, 'Team2_FIFA_RANK', pred_set['Team2'].map(last_team_scores.set_index('team')['rank'])) pred_set.insert(4, 'Team1_Goalkeeper_Score', pred_set['Team1'].map(last_team_scores.set_index('team')['goalkeeper_score'])) pred_set.insert(5, 'Team2_Goalkeeper_Score', pred_set['Team2'].map(last_team_scores.set_index('team')['goalkeeper_score'])) pred_set.insert(6, 'Team1_Defense', pred_set['Team1'].map(last_team_scores.set_index('team')['defense_score'])) pred_set.insert(7, 'Team1_Offense', pred_set['Team1'].map(last_team_scores.set_index('team')['offense_score'])) pred_set.insert(8, 'Team1_Midfield', pred_set['Team1'].map(last_team_scores.set_index('team')['midfield_score'])) pred_set.insert(9, 'Team2_Defense', pred_set['Team2'].map(last_team_scores.set_index('team')['defense_score'])) pred_set.insert(10, 'Team2_Offense', pred_set['Team2'].map(last_team_scores.set_index('team')['offense_score'])) pred_set.insert(11, 'Team2_Midfield', pred_set['Team2'].map(last_team_scores.set_index('team')['midfield_score'])) return pred_set