pandas - 如何制作新的 0 和 1 列？

Question

我有一个 pandas 数据框，我想用 0 和 1 创建一个新列:

如果 col1 为零且 col2 为正，则将新列设置为 1。如果 'col1 为零且 col2 为负，则将新列设置为 0。如果 col1 为 1 且col2 为正，则设置新列为 0。如果 'col1 为 1 且 col2 为负，则将新列设置为 1。

col1   col2
0       2
0       -4
1       -2
1       5
1       9

new_colum
1
0
1 
0
0

Answer 1

您可以确定 col2 是否为正并获得与 col1 的绝对差值( bool 值的行为类似于 0/1):

df['new_column'] = df['col1'].sub(df['col2'].gt(0)).abs()

或者，比较两个输出，您希望它们不同:

df['new_column'] = df['col1'].ne(df['col2'].gt(0)).astype(int)

输出:

   col1  col2  new_column
0     0     2           1
1     0    -4           0
2     1    -2           1
3     1     5           0
4     1     9           0