如何自定义特定熊猫数据框列的函数调用顺序？

Question

我有一个自定义函数列表：

Func =[{'func':'push','col':'A'},{'func':'pull','col':'B'}]

def push : #do something
def pull : #do something

我的数据框：

id  A   B
1  23  24
2  99  88

我希望根据上面提到的列表“Func”，按照列表中给出的特定顺序对该数据帧进行操作。 即，首先在 columnA 上调用函数“Push”，获取该输出数据帧，然后在新数据帧上调用函数“Pull”。

Answer 1

由于函数是 python 中的一等公民，您可以将它们作为值放在 dict 中，

然后只需迭代您的列表并在所需列上调用函数：

编辑：现在使用 exec 将函数名称作为字符串（请参阅 OP 评论）

import pandas as pd

df = pd.DataFrame({"A":[23,99], "B":[24,88]})

def push(series):
  series -= 1
def pull(series):
  series /= 2

print("before:")
print(df)

Func =[{'func':'push','col':'A'},{'func':'pull','col':'B'}]

for f in Func:
  exec('{}(df[f["col"]])'.format(f['func']))

print("after:")
print(df)

输出：

before:
    A   B
0  23  24
1  99  88
after:
    A     B
0  22  12.0
1  98  44.0