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

时间:2023-03-08 16:25:00
Python中将函数作为另一个函数的参数传入并调用

在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)