pandas使用apply函数

# coding=utf-8
import pandas as pd

# 创建一个示例DataFrame
df = pd.DataFrame({
    'A': [1, 2, 3],
    'B': [4, 5, 6]
})


# 定义一个函数，计算两列的和
def sum_two_columns(row):
    return row['A'] + row['B']


# 使用apply函数
df['Sum'] = df.apply(sum_two_columns, axis=1)
print(df)