要将两列合并为一个复合列,可以使用pandas的DataFrame
的apply
方法,传入一个自定义函数,该函数用来将两列的值合并为一个复合值。
下面是一个示例代码:
import pandas as pd
# 创建一个包含两列的DataFrame
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)
# 自定义函数,将两列的值合并为一个复合值
def combine_columns(row):
return str(row['A']) + str(row['B'])
# 使用apply方法将两列合并为一个复合列
df['C'] = df.apply(combine_columns, axis=1)
print(df)
输出结果:
A B C
0 1 4 14
1 2 5 25
2 3 6 36
在上面的代码中,首先创建了一个包含两列的DataFrame。然后定义了一个自定义函数combine_columns
,该函数接受一个行数据作为参数,并将该行的’A’列和’B’列的值合并为一个复合值。最后,使用apply
方法将该函数应用到每一行数据上,并将结果赋值给新的列’C’。