函数调用中的星号

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()，这会将它们链接在一起，但您没有单独的列表变量，因此您可以使用 * 运算符将列表列表展开为多个列表参数。

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]