python - 根据 Pandas 中的第三列保留两列之间的值

Question

我有三列，A、B 和 C。我想根据 C 的值创建第四列 D，其中包含 A 或 B 的值。例如:

   A   B   C   D 
0  1   2   1   1
1  2   3   0   3
2  3   4   0   4
3  4   5   1   4

在上面的示例中，如果 C 的值为 1，则 D 列采用 A 列的值；如果 C 的值为 0，则采用 B 列的值。在 Pandas 中是否有一种优雅的方法可以做到这一点？谢谢您的帮助。

Answer 1

使用numpy.where :

In [20]: df
Out[20]:
   A  B  C
0  1  2  1
1  2  3  0
2  3  4  0
3  4  5  1

In [21]: df['D'] = np.where(df.C, df.A, df.B)

In [22]: df
Out[22]:
   A  B  C  D
0  1  2  1  1
1  2  3  0  3
2  3  4  0  4
3  4  5  1  4