函数调用中的星号

Question

我正在使用itertools.chain以这种方式"压扁"列表列表:

uniqueCrossTabs = list(itertools.chain(*uniqueCrossTabs))

这有什么不同于说:

uniqueCrossTabs = list(itertools.chain(uniqueCrossTabs))

Answer 1

* 是"splat"运算符:它将列表作为输入,并将其扩展为函数调用中的实际位置参数.

如果uniqueCrossTabs是的话[ [ 1, 2 ], [ 3, 4 ] ],那itertools.chain(*uniqueCrossTabs)就像说的一样itertools.chain([ 1, 2 ], [ 3, 4 ])

这明显不同于传入uniqueCrossTabs.在您的情况下,您有一个您希望展平的列表列表; 什么itertools.chain()是在你传递给它的所有位置参数的串联上返回一个迭代器,其中每个位置参数本身是可迭代的.

换句话说,您希望将每个列表uniqueCrossTabs作为参数传递给chain()它们,它们将它们链接在一起,但是您没有单独变量中的列表,因此您使用*运算符将列表列表扩展为多个列表参数.

正如Jochen Ritzel在评论中指出的那样,chain.from_iterable()它更适合于这种操作,因为它假定可以迭代一次迭代.您的代码变得简单:

uniqueCrossTabs = list(itertools.chain.from_iterable(uniqueCrossTabs))

Answer 2

它将序列拆分为函数调用的单独参数.

>>> def foo(a, b=None, c=None):
...   print a, b, c
... 
>>> foo([1, 2, 3])
[1, 2, 3] None None
>>> foo(*[1, 2, 3])
1 2 3
>>> def bar(*a):
...   print a
... 
>>> bar([1, 2, 3])
([1, 2, 3],)
>>> bar(*[1, 2, 3])
(1, 2, 3)

Answer 3

只是解释概念/使用它的另一种方式.

import random

def arbitrary():
    return [x for x in range(1, random.randint(3,10))]

a, b, *rest = arbitrary()

# a = 1
# b = 2
# rest = [3,4,5]