下面是一个示例代码,它比较两列并用数字替换NaN:
import pandas as pd
import numpy as np
# 创建一个示例数据集
data = {'A': [1, np.nan, 3, 4, 5],
'B': [np.nan, 2, 3, np.nan, 6]}
df = pd.DataFrame(data)
# 比较两列并用数字替换NaN
df['C'] = np.where(pd.isnull(df['A']), df['B'], df['A'])
print(df)
运行上面的代码,输出结果如下:
A B C
0 1.0 NaN 1.0
1 NaN 2.0 2.0
2 3.0 3.0 3.0
3 4.0 NaN 4.0
4 5.0 6.0 5.0
在这个示例中,我们使用了np.where
函数来比较两列的值。如果列'A'的值为NaN,则将列'B'的值放入新列'C'中,否则将列'A'的值放入新列'C'中。
上一篇:比较两列并突出显示指定值不可用
下一篇:比较两列并在新列中添加一个值