Pandas:我如何将一个列名传递给一个函数，然后在'apply‘中使用？

import pandas as pd

# 定义一个函数，对传入的列进行操作
def my_function(column):
    # 在这里进行你的操作，例如对列进行加倍
    return column * 2

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

# 使用apply函数将函数应用到指定的列上
df['A'] = df['A'].apply(my_function)

print(df)

   A  B
0  2  4
1  4  5
2  6  6