2022年 11月 4日

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

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

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

示例代码：

  1. def func_a(func, *args, **kwargs):
  2.  
  3.   print(func(*args, **kwargs))
  4.  
  5.  
  6. def func_b(*args):
  7.  
  8.   return args
  9.  
  10.  
  11. if __name__ == '__main__':
  12.  
  13.   func_a(func_b, 1, 2, 3)
  14.  
  15. Output:
  16. ----------
  17. (1, 2, 3)
  18. ----------

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

  1. def func_a(arg_a, func, **kwargs):
  2.  
  3.   print(arg_a)
  4.  
  5.   print(func(**kwargs))
  6.  
  7.  
  8. def func_b(arg_a):
  9.  
  10.   print(arg_a)
  11.  
  12.  
  13. if __name__ == '__main__':
  14.  
  15.   func_a(arg_a='Hello Python', func=func_b)
  16.  
  17.  
  18. Output:
  19. --------------------------------------------------------------------
  20. TypeError: func_b() missing 1 required positional argument: 'arg_a'
  21. --------------------------------------------------------------------

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

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

  1. def func_a(arg_a, func, **kwargs):
  2.  
  3.   print(arg_a)
  4.  
  5.   func(arg_a=arg_a, **kwargs)
  6.  
  7.  
  8. def func_b(arg_a):
  9.  
  10.   print(arg_a)
  11.  
  12.  
  13. if __name__ == '__main__':
  14.  
  15.   func_a(arg_a='Hello Python', func=func_b)
  16.  
  17.  
  18. Output:
  19. --------------
  20. Hello Python
  21. Hello Python
  22. --------------

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

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