2022年 11月 4日

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

参考  在Python中将函数作为另一个函数的参数传入并调用的方法 – 云+社区 – 腾讯云

在Python中,函数本身也是对象,所以可以将函数作为参数传入另一函数并进行调用,以节省代码量。在旧版本中,可以使用apply(function, *args, **kwargs)进行调用,但是在新版本中已经移除,以function(*args, **kwargs)进行替代,所以也不应该再使用apply方法

示例代码:

  1. def func_a(func, *args, **kwargs):
  2. print(func(*args, **kwargs))
  3. def func_b(*args):
  4. return args
  5. if __name__ == '__main__':
  6. func_a(func_b, 1, 2, 3)
  7. Output:
  8. ----------
  9. (1, 2, 3)
  10. ----------

在代码中,将函数func_b作为函数func_a的参数传入,将函数func_b的参数以元组args传入,并在调用func_b时,作为func_b的参数。但是这里存在一个问题,但func_a和func_b需要同名的参数时,就会出现异常,如:

  1. def func_a(arg_a, func, **kwargs):
  2. print(arg_a)
  3. print(func(**kwargs))
  4. def func_b(arg_a):
  5. print(arg_a)
  6. if __name__ == '__main__':
  7. func_a(arg_a='Hello Python', func=func_b)
  8. Output:
  9. --------------------------------------------------------------------
  10. TypeError: func_b() missing 1 required positional argument: 'arg_a'
  11. --------------------------------------------------------------------

虽然通过修改,手动将arg_a作为参数传入func中进行调用,可以正常运行,但这明显不符合设计初衷:

在func_a中执行func(**kwargs)时,很可能并不知道func到底需要什么参数。换句话说,如果已经提前知道需要调用什么函数,那完全不必要把函数作为参数传入另一个函数并调用,直接调用函数即可。

  1. def func_a(arg_a, func, **kwargs):
  2. print(arg_a)
  3. func(arg_a=arg_a, **kwargs)
  4. def func_b(arg_a):
  5. print(arg_a)
  6. if __name__ == '__main__':
  7. func_a(arg_a='Hello Python', func=func_b)
  8. Output:
  9. --------------
  10. Hello Python
  11. Hello Python
  12. --------------

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

  1. def func_a(arg_a, func, **kwargs):
  2. print(arg_a)
  3. func(arg_a=arg_a, **kwargs)
  4. def func_b(arg_a):
  5. print(arg_a)
  6. def func_c():
  7. print('Hello World')
  8. if __name__ == '__main__':
  9. func_a(arg_a='Hello Python', func=func_b)
  10. func_a(arg_a='Hello Python', func=func_c)
  11. Output:
  12. ------------------------------------------------------------------
  13. Hello Python
  14. Hello Python
  15. Hello Python
  16. Traceback (most recent call last):
  17. File "D:/pylearning/config/config.py", line 22, in <module>
  18. func_a(arg_a='Hello Python', func=func_c)
  19. File "D:/pylearning/config/config.py", line 5, in func_a
  20. func(arg_a=arg_a, **kwargs)
  21. TypeError: func_c() got an unexpected keyword argument 'arg_a'
  22. -----------------------------------------------------------------