Python数据处理类中如何预填充函数参数并构建待输入管道？

问题背景 #

我正在编写一个数据处理类A，包含以下方法：

class A:
    def func_1(self, input, param_1, param_2):
        # some logics
        return res

    def func_2(self, input, param_3, param_4, param_5):
        # some logics
        return res

    def func_3(self, input, param_6, param_7):
        # some logics
        return res
    
    def pipeline(self, config, input ...):
        # 如何实现这里？

希望构建pipeline方法，从配置文件读取参数后，生成一个预填充好固定参数、仅等待输入数据的函数列表，之后可以遍历输入数据，依次调用列表中的函数处理，预期效果如下：

# 预期生成的函数列表
pipe = [预填充param1/2的func_1, 预填充param3/4/5的func_2, 预填充param6/7的func_3]

# 遍历输入处理
for i in inputs:
    i = pipe[0](input=i)
    i = pipe[1](input=i)
    i = pipe[2](input=i)
    print(i)

想知道是否可以实现这种“预编译”效果，还是必须为每个输入重构函数？

实现方案 #

完全可以实现，不需要为每个输入重构函数。以下是两种简单可行的实现方式：

方式1：使用functools.partial #

Python标准库的functools.partial可以绑定函数的固定参数，返回一个新的可调用对象，仅需传入剩余的input参数即可。

完整代码示例：

from functools import partial

class A:
    def func_1(self, input, param_1, param_2):
        # 示例逻辑：返回输入加两个参数的和
        return input + param_1 + param_2

    def func_2(self, input, param_3, param_4, param_5):
        # 示例逻辑：返回输入乘三个参数的积
        return input * param_3 * param_4 * param_5

    def func_3(self, input, param_6, param_7):
        # 示例逻辑：返回输入减两个参数的差
        return input - param_6 - param_7
    
    def pipeline(self, config):
        # 绑定固定参数，生成预配置函数列表
        pipe = [
            partial(self.func_1, param_1=config.param_1, param_2=config.param_2),
            partial(self.func_2, param_3=config.param_3, param_4=config.param_4, param_5=config.param_5),
            partial(self.func_3, param_6=config.param_6, param_7=config.param_7)
        ]
        return pipe

# 模拟配置类
class Config:
    def __init__(self):
        self.param_1 = 1
        self.param_2 = 2
        self.param_3 = 3
        self.param_4 = 4
        self.param_5 = 5
        self.param_6 = 6
        self.param_7 = 7

# 测试
if __name__ == "__main__":
    a = A()
    config = Config()
    pipe = a.pipeline(config)
    
    inputs = [10, 20, 30]
    for i in inputs:
        current = i
        for func in pipe:
            current = func(input=current)
        print(current)

方式2：使用lambda表达式 #

如果不想导入额外库，也可以用lambda直接封装，效果一致：

class A:
    # 省略func_1/func_2/func_3定义，同上
    
    def pipeline(self, config):
        pipe = [
            lambda x: self.func_1(x, param_1=config.param_1, param_2=config.param_2),
            lambda x: self.func_2(x, param_3=config.param_3, param_4=config.param_4, param_5=config.param_5),
            lambda x: self.func_3(x, param_6=config.param_6, param_7=config.param_7)
        ]
        return pipe

# 测试逻辑同上

注意事项 #

• 使用lambda时要注意变量捕获问题：如果在循环中批量生成lambda，可能因延迟绑定导致参数错误，但这里是逐个显式定义，无需担心。
• partial会保留原函数的元信息（如函数名），调试时更友好；lambda则更轻量简洁。

内容的提问来源于stack exchange，提问作者A.Y