我们可以使用pandas库中的DataFrame对象提供的assign()方法来实现此功能。该方法可以根据给定的参数和计算公式创建新的列。
下面是示例代码:
import pandas as pd
# 定义参数
param1 = 5
param2 = "hello"
# 创建DataFrame对象
df = pd.DataFrame({"A": [1, 2, 3], "B": [4, 5, 6]})
# 定义函数,将参数和DataFrame对象作为输入
def add_new_column(df, param1, param2):
# 计算公式
new_col = df["A"] * param1 + param2
# 使用assign()方法创建新列
df = df.assign(C=new_col)
return df
# 调用函数并输出结果
df = add_new_column(df, param1, param2)
print(df)
在上面的代码中,我们通过调用assign()方法创建了一个名为“C”的新列,该列的值根据输入参数和计算公式进行计算。最终我们可以看到,输出结果中多了一列,该列是按照我们的要求来创建的。