Skip to main content
 首页 » 编程设计

python之如何在 Pandas 数据框中将两列合并为一列

2025年05月04日14daizhj

<分区>

想在特定条件下将两列合并为一列。
我的数据如下

idx    A    B 
0      0.5  1.2 
1      1.1  0.7 
2      0.1  0.3 
3      2.0  0.9 

使用 Pandas 数据框,我想在条件中创建 C 列。

  1. 如果 A 列 > B 列,C 列获取 A 列的值。
  2. 如果 B 列 > A 列,C 列获取 B 列的值。

在结果中,我希望这样。

idx    A    B    C 
0      0.5  1.2  1.2(B data) 
1      1.1  0.7  1.1(A data) 
2      0.1  0.3  0.3(B data) 
3      2.0  0.9  2.0(A data) 

我尝试了 .loc 函数,例如:
df['C'] = df.loc[df['A'] > df['B'], 'A']

但我无法替换或修改 C 列中的 Nan 值。

谢谢。