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

时间:2018-07-09 10:20:47

标签: python pandas

我有这个熊猫数据框:

enter image description here

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

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

我了解Pandas中的map和apply方法。但是我不确定它们是否允许像上面这样的条件表达式。

2 个答案:

答案 0 :(得分:3)

使用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)

答案 1 :(得分:0)

您可以尝试

/react-native-images-collage/package.json