pandas使用apply方法并发送列名吗？

import pandas as pd

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

# 自定义函数，接收一行数据和列名作为参数
def custom_func(row, col_name):
    return row[col_name] * 2

# 使用apply方法并发送列名
df['A_doubled'] = df.apply(custom_func, args=('A',), axis=1)
df['B_doubled'] = df.apply(custom_func, args=('B',), axis=1)

print(df)

   A  B  A_doubled  B_doubled
0  1  4          2          8
1  2  5          4         10
2  3  6          6         12