python - 是否可以使用 Python 中的 Pandas 获取任何数据框和任何列作为其输入的函数？

Question

假设我们有一个这样的数据框:

 my_df>>  

  column_1   column_2 column_3  column_4
0  0.276162  0.552951  0.866023  0.571535
1  0.112933  0.549487  0.626958  0.988705
2  0.916932  0.561641  0.220696  0.545019

我能否有一个函数可以获取像这样的任何数据框并将其任何列作为其输入？

为了更好地说明，如果我有这样的功能:

def multiply_5(df,column):
    df.column=df.column.apply(lambda x:x*5-3)

如果我像这样使用它，它会起作用吗？ :

multiply_5(my_df,column_2)

为了得到这个:

my_df.column_2=my_df.column_2.apply(lambda x:x*5-3)

我知道我写的这种特定方式行不通，但是有什么简单的方法可以将这个函数用于其他数据框吗？

Answer 1

尝试:

def multiply_5(df,column):
    df[column]=df[column].apply(lambda x:x*5-3)
    return df

df = multiply_5(df, "column_2")

print(df)
    column_1    column_2    column_3    column_4
0   0.276162    -4.176225   0.866023    0.571535
1   0.112933    -4.262825   0.626958    0.988705
2   0.916932    -3.958975   0.220696    0.545019

甚至更奇怪的方式通过属性做同样的事情:

def multiply_5(df,column):
    setattr(df,column, getattr(df,column).apply(lambda x:x*5-3))
    return df