如何将函数与if语句同时应用于多个数据框列

# 导入所需的库
import pandas as pd

# 创建一个示例数据框
df = pd.DataFrame({'A': [1, 2, 3, 4],
                   'B': [5, 6, 7, 8],
                   'C': [9, 10, 11, 12]})

# 定义一个函数，根据条件进行计算
def calculate(x):
    if x < 5:
        return x * 2
    else:
        return x * 3

# 使用apply函数将函数应用于多个列
df[['A', 'B', 'C']] = df[['A', 'B', 'C']].applymap(calculate)

# 输出结果
print(df)