温馨提示×

pandas怎么合并两列为一个复合列

小亿
273
2023-12-07 14:13:33
栏目: 编程语言

要将两列合并为一个复合列,可以使用pandas的DataFrameapply方法,传入一个自定义函数,该函数用来将两列的值合并为一个复合值。

下面是一个示例代码:

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’。

0