pandas.DataFrame.apply()
, .
:
def get_team_score(team):
def f(row):
if row.Team1 == team:
return row['Team1 Score']
if row.Team2 == team:
return row['Team2 Score']
return f
:
from io import StringIO
df = pd.read_csv(data)
print(df)
print(df.apply(get_team_score('Milan'), axis=1).values)
:
import pandas as pd
data = StringIO(u"""Date,Team1,Team2,Team1 Score,Team2 Score,Event
8/2/17,Juventus,Milan,2,1,Friendly match
6/2/17,Milan,Napoli,3,0,Friendly match
5/1/17,Milan,Sampdoria,1,0,Friendly match
25/12/16,Parma,Milan,0,5,Friendly match
""")
:
Date Team1 Team2 Team1 Score Team2 Score Event
0 8/2/17 Juventus Milan 2 1 Friendly match
1 6/2/17 Milan Napoli 3 0 Friendly match
2 5/1/17 Milan Sampdoria 1 0 Friendly match
3 25/12/16 Parma Milan 0 5 Friendly match
[1 3 1 5]
source
share