<p>检查此代码:</p>
<pre><code>import pandas as pd
df_in = pd.read_csv('data.csv')
df_out = pd.DataFrame(columns = ['team', 'opponent', 'games_played', 'wins', 'draws', 'losses', 'goals_scored', 'goals_conceded'])
team = 'Arsenal'
for index, row in df_in.iterrows():
if row['home_team'] == team:
opponent = row['away_team']
if row['home_goals'] > row['away_goals']:
win = 1
draw = 0
loss = 0
elif row['home_goals'] < row['away_goals']:
win = 0
draw = 0
loss = 1
else:
win = 0
draw = 1
loss = 0
goals_scored = row['home_goals']
goals_conceded = row['away_goals']
else:
opponent = row['home_team']
if row['home_goals'] > row['away_goals']:
win = 0
draw = 0
loss = 1
elif row['home_goals'] < row['away_goals']:
win = 1
draw = 0
loss = 0
else:
win = 0
draw = 1
loss = 0
goals_scored = row['away_goals']
goals_conceded = row['home_goals']
games_played = 1
if opponent not in df_out['opponent'].unique():
match = pd.DataFrame({'team': team,
'opponent': opponent,
'games_played': games_played,
'wins': win,
'draws': draw,
'losses': loss,
'goals_scored': goals_scored,
'goals_conceded': goals_conceded},
index = [0])
df_out = pd.concat([df_out, match], ignore_index = True)
else:
df_out.loc[df_out['opponent'] == opponent, 'games_played'] += games_played
df_out.loc[df_out['opponent'] == opponent, 'wins'] += win
df_out.loc[df_out['opponent'] == opponent, 'draws'] += draw
df_out.loc[df_out['opponent'] == opponent, 'losses'] += loss
df_out.loc[df_out['opponent'] == opponent, 'goals_scored'] += goals_scored
df_out.loc[df_out['opponent'] == opponent, 'goals_conceded'] += goals_conceded
</code></pre>
<p>此代码将以<code>df_in</code>的形式加载数据,并使用所需的数据创建一个<code>df_out</code>。<br/>
输出:</p>
<pre><code> team opponent games_played wins draws losses goals_scored goals_conceded
0 Arsenal Chelsea 3 1 1 1 9 7
1 Arsenal ManCity 2 0 1 1 2 3
</code></pre>