Python中函数参数传递方法*args, **kwargs,还有其他

liftword2个月前 (03-17)技术文章7

来源:DeepHub IMBA本文约3000字,建议阅读9分钟本文将讨论Python的函数参数。


我们将了解*args和**kwargs,/和*的都是什么,虽然这个问题是一个基本的python问题,但是在我们写代码时会经常遇到,比如timm中就大量使用了这样的参数传递方式。



定义和传递参数


parameters 和arguments 之间的区别是什么?许多人交替使用这些术语,但它们是有区别的:


  • Parameters 是函数定义中定义的名称;
  • Arguments是传递给函数的值。



红色的是parameters , 绿色的是arguments。


传递参数的两种方式:


我们可以按位置和关键字传递参数。在下面的例子中,我们将值hello作为位置参数传递。值world 用关键字传递的:


 def the_func(greeting, thing):

   print(greeting + ' ' + thing)




 the_func('hello', thing='world')

位置参数和kwargs(关键字参数)之间的区别在于传递位置参数的顺序很重要。如果调用the_func('world', 'hello')它会打印world hello。传递kwargs的顺序并不重要:


 the_func('hello', 'world')                  # -> 'hello world'

 the_func('world', 'hello')                  # -> 'world hello'

 the_func(greeting='hello', thing='world')   # -> 'hello world'

 the_func(thing='world', greeting='hello')   # -> 'hello world'

 the_func('hello', thing='world')            # -> 'hello world'


只要kwarg在位置参数之后,就可以混合和匹配位置参数和关键字参数,以上就是我们在python教程中经常看到的内容,下面我们继续:


函数参数


我们将演示6个函数参数传递的方法,这些方法能够覆盖到所有的问题。


1. 如何获得所有未捕获的位置参数。使用*args,让它接收一个不指定数量的形参。


 def multiply(a, b, *args):

   result = a * b

   for arg in args:

     result = result * arg

   return result


在这个函数中,我们通常定义前两个参数(a和b)。然后使用*args将所有剩余参数打包到一个元组中。可以把*看作是获取到了其他没有处理的参数,并将它们收集到一个名为“args”的元组变量中:


 multiply(1, 2)          # returns 2

 multiply(1, 2, 3, 4)    # returns 24


最后一次调用将值1赋给参数a,将2赋给参数b,并将arg变量填充为(3,4)。由于这是一个元组,我们可以在函数中循环它并使用这些值进行乘法!


2. 如何获得所有未捕获的关键字参数。与*args类似,这次是两个星号**kwargs:


 def introduce(firstname, lastname, **kwargs):

   introduction = f"I am {firstname} {lastname}"

   for key, value in kwargs.items():

     introduction += f" my {key} is {value} "

   return introduction


**kwargs关键字会将所有不匹配的关键字参数存储在一个名为kwargs的字典中。然后可以像上面的函数一样访问这个字典。


 print(introduce(firstname='mike', lastname='huls'))

 # returns "I am mike huls"




 print(introduce(firstname='mike', lastname='huls', age=33, website='mikehuls.com'))

 # I am mike huls my age is 33 my website is overfit.cn


3. 如果想只接受关键字参数,那怎么设计。可以强制函数只接受关键字参数。


 def transfer_money(*, from_account:str, to_account:str, amount:int):

   print(f'Transfering ${amount} FORM {from_account} to {to_account}')




 transfer_money(from_account='1234', to_account='6578', amount=9999)

 # won't work: TypeError: transfer_money() takes 0 positional arguments but 1 positional argument (and 2 keyword-only arguments) were given

 transfer_money('1234', to_account='6578', amount=9999)

 # won't work: TypeError: transfer_money() takes 0 positional arguments but 3 were given

 transfer_money('1234', '6578', 9999)


在上面的函数中,*星号获得了了所有不匹配的位置参数,但是并没有一个变量来接受它,也就是被忽略了。


4. 如何设计函数只接受位置参数。下面是一个只允许位置参数的函数示例:


 def the_func(arg1:str, arg2:str, /):

   print(f'provided {arg1=}, {arg2=}')




 # These work:

 the_func('num1', 'num2')

 the_func('num2', 'num1')




 # won't work: TypeError: the_func() got some positional-only arguments passed as keyword arguments: 'arg1, arg2'

 the_func(arg1='num1', arg2='num2')

 # won't work: TypeError: the_func() got some positional-only arguments passed as keyword arguments: 'arg2'

 the_func('num1', arg2='num2')


函数定义中的/强制在它之前的所有参数都是位置参数。这并不意味着/后面的所有参数都必须是kwarg-only;这些可以是位置和关键字。


看到这个你肯定会想,为什么想要这个?这不会降低代码的可读性吗?,我也觉得你说的非常正确,当定义一个非常明确的函数时,不需要关键字参数来指定它的功能。例如:


 def exceeds_100_bytes(x, /) -> bool:

   return x.__sizeof__() > 100




 exceeds_100_bytes('a')      

 exceeds_100_bytes({'a'})


在这个例子中,正在检查'a'的内存大小是否超过100字节。因为这个x对于我们来说他的名字不重要,在调用函数的时候不需要指定x= ' a '。比如说我们最常用的len,如果你调用len(__obj=[]) 这样看起来是不是有点呆萌,因为len是这么定义的:


def len(__obj: Sized) -> int:


5. 混合和匹配。作为一个例子,我们将看看前面讨论过的len函数。这个函数只允许位置参数。我们将通过允许开发人员选择是否计算重复项来扩展此函数,比如用kwargs传递这个关键字:


 def len_new(x, /, *, no_duplicates=False):

   if (no_duplicates):

     return len(list(set([a for a in x])))

   return len(x)


想计算变量x的len,只能按位置传递x形参的参数,因为它前面有一个/。no_duplicate参数必须与关键字一起传递,因为它跟在*后面。让我们看看这个函数都可以怎么调用:


 print(len_new('aabbcc'))                                  # returns 6

 print(len_new('aabbcc', no_duplicates=True))              # returns 3

 print(len_new([1, 1, 2, 2, 3, 3], no_duplicates=False))   # returns 6

 print(len_new([1, 1, 2, 2, 3, 3], no_duplicates=True))    # returns 3




 # Won't work: TypeError: len_() got some positional-only arguments passed as keyword arguments: 'x'

 print(len_new(x=[1, 1, 2, 2, 3, 3]))

 # Won't work: TypeError: len_new() takes 1 positional argument but 2 were given

 print(len_new([1, 1, 2, 2, 3, 3], True))


6. 最后把它们合在一起。下面的函数是一个非常极端的例子,说明了如何组合前面讨论的所有技术:它强制前两个参数以位置方式传递,接下来的两个参数可以以位置方式传递,并且带有关键字,然后是两个只有关键字的参数,然后我们用**kwargs捕获剩下的未捕获的参数。


 def the_func(pos_only1, pos_only2, /, pos_or_kw1, pos_or_kw2, *, kw1, kw2, **extra_kw):

   # cannot be passed kwarg   <-- --> can be passed 2 ways | --> can only be passed by kwarg

   print(f"{pos_only1=}, {pos_only2=}, {pos_or_kw1=}, {pos_or_kw2=}, {kw1=}, {kw2=}, {extra_kw=}")


调用方式如下:


 # works (pos_or_kw1 & pow_or_k2 can be passed positionally and by kwarg)

 pos_only1='pos1', pos_only2='pos2', pos_or_kw1='pk1', pos_or_kw2='pk2', kw1='kw1', kw2='kw2', extra_kw={}

 pos_only1='pos1', pos_only2='pos2', pos_or_kw1='pk1', pos_or_kw2='pk2', kw1='kw1', kw2='kw2', extra_kw={}

 pos_only1='pos1', pos_only2='pos2', pos_or_kw1='pk1', pos_or_kw2='pk2', kw1='kw1', kw2='kw2', extra_kw={'kw_extra1': 'extra_kw1'}




 # doesnt work, (pos1 and pos2 cannot be passed with kwarg)

 # the_func(pos_only1='pos1', pos_only2='pos2', pos_or_kw1='pk1', pos_or_kw2='pk2', kw1='kw1', kw2='kw2')




 # doesnt work, (kw1 and kw2 cannot be passed positionally)

 # the_func('pos1', 'pos2', 'pk1', 'pk2', 'kw1', 'kw2')


总结


看着很乱是吧,这就对了。因为python在设计时是一个很宽松的语言,并没有那么多的规范,用的人越多使用方法就越多,就变成了这样。


那么回到第一张图:


 def func(x,/,y,*,z,**k):

(x,/,y,*,z,**k):是函数的参数。总共有四个参数:


  • x: 是一个常规参数,这意味着它可以按位置传递,也可以按关键字传递。
  • /,: 是一个参数分隔符,将仅限位置的参数与其他参数分开。与前面的x结合,意味着x只能按位置传递。
  • y: 时另一个常规参数。
  • *: 是一个参数分隔符,用于分隔仅限位置参数和仅限关键字参数。它意味着后面的z只能通过关键字传递。
  • z: 是一个仅限关键字的参数。
  • **k: 这是一个参数,将所有剩余的关键字参数收集到一个名为' k '的字典中。


这样解释是不是就很明白了。


我们今天介绍的这个例子虽然在看源代码时没有遇到这么复杂的情况,但是在面试的时候还真有人问(虽然我觉得没啥用),所以最好还是知道一些,以免尴尬。


如果你忘记了,这里可以教你一个变通的办法,可以使用类似的回答:


上面的参数传递在开发时并不常用,因为对于开发规范来说,应该保证代码的可读性,我们这边遵循的开发规范是:

1、尽量不要在函数定义中将可变位置参数 *args 和可变关键字参数 **kwargs 放在一起,因为这样会让函数的调用方式变得不太直观。

2、在使用可变参数时,要保证函数的行为是可预测的。上面函数中的进行了太多的python语法糖,对于理解该函数的参数会造成很大的困惑,也就是可读性太差,我们在进行codereview(如果你了解什么是codereview就说,不了解就说组长检查)/组长merge代码 时会直接要求返工,所以我们在实际开发时是不会用这个的。

对于我阅读的开源代码,也都基本上使用的是 **kwargs 这种情况(这里可以举两个例子),还没有看到有人写这么乱的代码,我想要是写这样的代码估计开源的人也会被人吐糟(这里自己可以自行延伸),所以这些参数传递的规则我在学习的时候看到过,但是实际中没见过真正使用,就不太记住了。


回到本文,我们介绍了设计函数参数的所有方法,并了解了如何混合和匹配它们,虽然后面几个内容可能你一辈子也不会用到,但是了解一下也是好的,因为万一呢。

相关文章

Python 函数参数

1.函数基本参数#基本样例 函数参数的定义要和调用参数匹配,否则会报错 def fun01(a, b): print(a, b) if __name__ == '__main__':...

真相!Python 默认值参数深度解析,从踩坑到精通的进阶秘籍

一、开发中遇到的问题问题 1:默认值参数为可变对象的陷阱在 Python 开发里,不少初学者会在默认值参数为可变对象时踩坑。比如,我们想定义一个函数,用于向列表中添加元素,代码如下:def add_t...

Python 函数之参数

定义一个函数非常简单,但是怎么定义一个函数,需要什么参数,怎么去调用却是我们需要去思考的问题。如同大多数语言一样(如 Java),Python 也提供了多种参数的设定(如:默认值参数、关键字参数、形参...

Python命令行参数解析模块argparse

前言以前 optparse 比较火,但是在python2.7版本后,官方不再维护这个模块,转而大力支持 argparse 。argparse 模块可以让人轻松编写用户友好的命令行接口。她可以从 sys...

python函数参数详解

python有两种传参的方式,位置参数和关键字参数位置参数位置参数是基于它们的位置或顺序传递给函数的参数。这些参数是必需的,并且必须按照与函数参数列表匹配的正确顺序提供。以下是一个示例:def gre...

Python教程(25)——Python中参数类型详解

当我们在编写函数时,会定义一些占位符,这些占位符就是参数,参数是函数定义中用于接收外部传递值的占位符,这个会帮助我们在函数被调用时接收外部传递的值。在Python当中,有各种各样类型的参数,主要有位置...