使用Pandas库,可以使用“apply”函数来实现该功能。
示例代码:
import pandas as pd
# 创建示例数据
data = {'a': [1, 2, 3], 'b': [1, 4, 3]}
df = pd.DataFrame(data)
# 定义条件函数
def compare_columns(row):
if row['a'] == row['b']:
return row['a']
else:
return row['b']
# 添加新列
df['new_column'] = df.apply(compare_columns, axis=1)
# 打印结果
print(df)
输出结果:
a b new_column
0 1 1 1
1 2 4 4
2 3 3 3
可以看到,新的“new_column”列中,第一行的值为1(a和b列相同),第二行的值为4(a和b列不同)。第三行的值同样为3(a和b列相同)。