理解 Python 中的 *args 和 **kw

反复在Python的函数声明中见到*args和**kw，对其作用非常模糊，所以特作整理。

Python是支持可变参数的，最简单的方法莫过于使用默认参数，例如：

1
2
3
4
5
6
7
8
9
10
11
12
13
def test_defargs(one, two = 2):
   print 'Required argument: ', one
   print 'Optional argument: ', two
test_defargs(1)
# result:
# Required argument: 1
# Optional argument: 2
test_defargs(1, 3)
# result:
# Required argument: 1
# Optional argument: 3

当然，本文章的主题并不是讲默认参数，而是另外一种达到可变参数 (Variable Argument) 的方法：使用*args和**kw语法。其中，*args是可变的positional arguments列表，**kw是可变的keyword arguments列表。并且，*args必须位于**kw之前，因为positional arguments必须位于keyword arguments之前。

首先介绍两者的基本用法。

下面一个例子使用*args，同时包含一个必须的参数：

1
2
3
4
5
6
7
8
9
10
11
def test_args(first, *args):
   print 'Required argument: ', first
   for v in args:
      print 'Optional argument: ', v
test_args(1, 2, 3, 4)
# result:
# Required argument: 1
# Optional argument:  2
# Optional argument:  3
# Optional argument:  4

下面一个例子使用**kw, 同时包含一个必须的参数和*args列表：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
def test_kw(first, *args, **kw):
   print 'Required argument: ', first
   for v in args:
      print 'Optional argument (*args): ', v
   for k, v in kw.items():
      print 'Optional argument %s (*kw): %s' % (k, v)
test_kw(1, 2, 3, 4, k1=5, k2=6)
# results:
# Required argument:  1
# Optional argument (*args):  2
# Optional argument (*args):  3
# Optional argument (*args):  4
# Optional argument k2 (*kw): 6
# Optional argument k1 (*kw): 5

*args和**kw语法不仅可以在函数定义中使用，同样可以在函数调用的时候使用。不同的是，如果说在函数定义的位置使用*args和**kw是一个将参数pack的过程，那么在函数调用的时候就是一个将参数unpack的过程了。下面使用一个例子来加深理解：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
def test_args(first, second, third, fourth, fifth):
    print 'First argument: ', first
    print 'Second argument: ', second
    print 'Third argument: ', third
    print 'Fourth argument: ', fourth
    print 'Fifth argument: ', fifth
# Use *args
args = [1, 2, 3, 4, 5]
test_args(*args)
# results:
# First argument:  1
# Second argument:  2
# Third argument:  3
# Fourth argument:  4
# Fifth argument:  5
# Use **kw
kw = {
    'first': 1,
    'second': 2,
    'third': 3,
    'fourth': 4,
    'fifth': 5
}
test_args(**kw)
# results:
# First argument:  1
# Second argument:  2
# Third argument:  3
# Fourth argument:  4
# Fifth argument:  5

使用*args和**kw可以非常方便的定义函数，同时可以加强扩展性，以便日后的代码维护。

文章转自：理解 Python 中的args和*kw

推荐文章