Setting value of a column based on values of other columns in Pandas dataframe

Question

I have this Pandas dataframe:

This is some fictional data about games of cricket played between two countries. I want to set the value of column winner based on this simple logic:

if country_1_runs == country_2_runs:
    winner = 3
elif country_1_runs > country_2_runs:
    winner = 1
else:
    winner = 2

I know about map and apply methods in Pandas. But I'm not certain if they allow conditional expressions like the above.

jezrael · Accepted Answer · 2018-07-09 10:23:28Z

4

m1 = df.country_1_runs == df.country_2_runs
m2 = df.country_1_runs > df.country_2_runs

df['winner'] = np.select([m1, m2], [3, 1], default=2)

answered Jul 9, 2018 at 10:23

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

This is a really elegant solution.

@RodrikTheReader - Yes, I agree.

Sreekiran A R · Accepted Answer · 2018-07-09 10:28:26Z

0

You can try

df['winner'] = np.where(df['country_1_runs'] == df['country_2_runs'], 3, 
    np.where(df['country_1_runs']> df['country_2_runs'], 1, 2))

answered Jul 9, 2018 at 10:28

Sreekiran A R

3,4253 gold badges25 silver badges45 bronze badges