python - 函数调用中的星号

我正在使用itertools.chain以这种方式“拉平”列表列表：

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

这跟说的有什么不同？

uniqueCrossTabs = list(itertools.chain(uniqueCrossTabs))

最佳答案

*是“ 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))