内容
隐藏
问题:函数调用中的星号
我正在使用itertools.chain以这种方式“拉平”列表列表:
uniqueCrossTabs = list(itertools.chain(*uniqueCrossTabs))
这跟说的有什么不同?
uniqueCrossTabs = list(itertools.chain(uniqueCrossTabs))
回答 0
*
是“ splat”运算符:它接受一个列表作为输入,并将其扩展为函数调用中的实际位置参数。
所以如果uniqueCrossTabs
是[ [ 1, 2 ], [ 3, 4 ] ]
,那就itertools.chain(*uniqueCrossTabs)
等于说itertools.chain([ 1, 2 ], [ 3, 4 ])
这与传递just显然不同uniqueCrossTabs
。对于您的情况,您有一个想要拼合的列表列表;什么itertools.chain()
确实是在所有你传递给它的位置参数,其中每个位置参数是在自己的权利迭代拼接返回一个迭代。
换句话说,您希望将每个列表uniqueCrossTabs
作为参数传递给chain()
,这会将它们链接在一起,但是您没有在单独的变量中使用列表,因此可以使用*
运算符将列表扩展为多个列表参数。
正如Jochen Ritzel在评论中指出的那样,chain.from_iterable()
它更适合于此操作,因为它假定一个可迭代的对象开始。然后,您的代码将变得简单:
uniqueCrossTabs = list(itertools.chain.from_iterable(uniqueCrossTabs))
回答 1
它将序列拆分为函数调用的单独参数。
>>> 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)
回答 2
只是解释概念/使用它的另一种方法。
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]