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

Output:
----------
(1, 2, 3)
----------

在代码中,将函数func_b作为函数func_a的参数传入,将函数func_b的参数以元组args传入,并在调用func_b时,作为func_b的参数。但是这里存在一个问题,但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)


Output:
--------------------------------------------------------------------
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)


Output:
--------------
Hello Python
Hello Python
--------------

当加入第三个函数,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)


Output:
------------------------------------------------------------------
Hello Python
Hello Python
Hello Python
Traceback (most recent call last):
  File "D:/pylearning/config/config.py", line 22, in <module>
    func_a(arg_a='Hello Python', func=func_c)
  File "D:/pylearning/config/config.py", line 5, in func_a
    func(arg_a=arg_a, **kwargs)
TypeError: func_c() got an unexpected keyword argument 'arg_a'

-----------------------------------------------------------------

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐