根据df中其他两个列的比较设置列值

时间:2020-01-08 17:25:25

标签: python python-3.x pandas dataframe

我正在尝试根据其他两列的比较将df列中的值设置为0或1。

我最初的df是这样的:

col1   col2    col3
1      3       NaN
5      1       NaN
7      4       NaN
5      10      NaN

我正在使用此代码尝试将col3中的值设置为1,其中col1大于col2,所有其他行均为0:

df[df['col1'] >= df['col2']]['col3'] = 1
df[df['col1'] < df['col2']]['col3'] = 0

这肯定是错误的方法,但是我不确定其他方法。

所需结果是具有以下值的df:

col1   col2    col3
1      3       0
5      1       1
7      4       1
5      10      0

先谢谢了。

1 个答案:

答案 0 :(得分:3)

您可以使用:

setlocale()