python - 根据 Pandas 数据框中其他列的值设置列的值

Question

我有这个 Pandas 数据框:

这是有关两国之间进行的板球比赛的一些虚构数据。我想根据这个简单的逻辑设置列获胜者的值:

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

我了解 Pandas 中的映射和应用方法。但我不确定他们是否允许像上面这样的条件表达式。

Answer 1

使用numpy.select :

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)