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

Question

我有这个熊猫数据框：

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

if country_1_runs == country_2_runs:

    winner = 3

elif country_1_runs > country_2_runs:

    winner = 1

else:

    winner = 2

我知道有关Pandas中的地图和应用方法的信息。但是我不确定它们是否允许像上面这样的条件表达式。

Answer 1

你可以试试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))