如何在panda中设置列值条件下的值

Question

这是我的数据帧

    col1    col2
0   1.0     5.0
1   2.0     6.0
2   NaN     7.0
3   3.0     8.0

我想添加具有值col1或col2作为值条件的列("l3"(。Col3值与col1相同，但如果col1值为"NaN"，则Col3值设置为col2。

我的预期输出是.

    col1    col2   col3
0   1.0     5.0    1.0
1   2.0     6.0    2.0
2   NaN     7.0    7.0
3   3.0     8.0    3.0

我该怎么做？

Answer 1

pandas有一个内置函数：combine_first

df["col3"] = df["col1"].combine_first(df["col2"])