Python中将函数作为另一个函数的参数传入并调用

Posted 2020-07-28 BlackMatrix

在Python中，函数本身也是对象，所以可以将函数作为参数传入另一函数并进行调用

在旧版本中，可以使用apply(function, *args, **kwargs)进行调用，但是在新版本中已经移除，以function(*args, **kwargs)进行替代，所以也不应该再使用apply方法

示例代码：

def func_a(func, *args, **kwargs):
    print(func(*args, **kwargs))

def func_b(*args):
    return args

if __name__ == ‘__main__‘:
    func_a(func_b, 1, 2, 3)

在代码中，将函数func_b作为函数func_a的参数传入，将函数func_b的参数以元组args传入，并在调用func_b时，作为func_b的参数。

运行结果：

(1, 2, 3)

 

但是这里存在一个问题，但func_a和func_b需要同名的参数时，就会出现异常，如：

def func_a(arg_a, func, **kwargs):
    print(arg_a)
    print(func(**kwargs))

def func_b(arg_a):
    print(arg_a)

if __name__ == ‘__main__‘:
    func_a(arg_a=‘Hello Python‘, func=func_b)

异常信息：

TypeError: func_b() missing 1 required positional argument: ‘arg_a‘

 

虽然通过修改，手动将arg_a作为参数传入func中进行调用，可以正常运行，但这明显不符合设计初衷：在func_a中执行func(**kwargs)时，很可能并不知道func到底需要什么参数。换句话说，如果已经提前知道需要调用什么函数，那完全不必要把函数作为参数传入另一个函数并调用，直接调用函数即可。

def func_a(arg_a, func, **kwargs):
    print(arg_a)
    func(arg_a=arg_a, **kwargs)

def func_b(arg_a):
    print(arg_a)

if __name__ == ‘__main__‘:
    func_a(arg_a=‘Hello Python‘, func=func_b)

当加入第三个函数，func_c，它不需要arg_a这个参数时，就会出现问题。

def func_a(arg_a, func, **kwargs):
    print(arg_a)
    func(arg_a=arg_a, **kwargs)

def func_b(arg_a):
    print(arg_a)

def func_c():
    print(‘Hello World‘)

if __name__ == ‘__main__‘:
    func_a(arg_a=‘Hello Python‘, func=func_b)
    func_a(arg_a=‘Hello Python‘, func=func_c)

异常结果：

TypeError: func_c() got an unexpected keyword argument ‘arg_a‘

 

目前想到的解决办法是尽量避免func_a存在跟其他函数相同的参数，比如把func_a的arg_a参数改成func_a_arg_a。

def func_a(func_a_arg_a, func, **kwargs):
    print(func_a_arg_a)
    func(**kwargs)

def func_b(arg_a):
    print(arg_a)

def func_c():
    print(‘Hello World‘)

if __name__ == ‘__main__‘:
    func_a(func_a_arg_a=‘temp‘, arg_a=‘Hello Python‘, func=func_b)
    func_a(func_a_arg_a=‘temp‘, func=func_c)