python - 根据指定的参数修改函数中的内部操作-python pandas

Question

我有一个数据框：

name  age  city zipcode
 Jon   22  CA   0002
 Raj   34  GG   3454
 ...   ..  ..   ....
 ...   ..  ..   ....

我想编写一个函数Preprocess(df,[params])以便预处理应该只发生在将在参数中传递的数据帧的列上。

例如。

def preprocess(df,params):
    name_list = df['name'].tolist()
    mean_age = df['age'].mean()
    city_list = df['city'].tolist()
    zipcode_list = df['zipcode'].tolist()

    return [name_list, mean_age, city_list, zipcode_list]

result = preprocess(df,['age','city'])

我正在寻找的是，如果我只传递年龄和城市，我希望预处理只发生在这两列

ie
mean_age = df['age'].mean()
city_list = df['city'].tolist()

一种方法是使用 if else/try-catch

我正在寻找解决这个问题的最佳方法或 Pythonic 方法

标签： pythonpandas