函数调用中的星号


111

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

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

这跟说的有什么不同?

uniqueCrossTabs = list(itertools.chain(uniqueCrossTabs))

8
查看Python文档中的解压缩参数列表以获取更多信息。

8
您还应该签出**运算符-它的作用*与使用关键字参数相同。
肖恩·维埃拉

Answers:


181

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

9
@larsmans:我认为这个词在Ruby世界中更流行,但是它似乎也可以被Python接受,因为我喜欢它,因为它很有趣;-)
Cameron

1
@larsmans:有趣!我一直认为它指的是将列表拆成参数列表的动作,而不是实际字符本身。
卡梅伦

1
也许字符串不是最好的例子,因为不是每个人都认为字符串是可迭代的。顺便说一句:而不是chain(*it)我写chain.from_iterable(it)
Jochen Ritzel 2011年

@Jochen:绝对正确,我将其改为使用数字。另外,我什至不知道from_iterable存在!我将添加到我的答案不久
卡梅伦

1
@Ramy:*仅用于将列表分解为函数的位置参数(因此,非常具体)。您可以for l in uniqueCrossTabs:对它们进行迭代。不幸的是,很难看到*它在工作,因为它仅在将列表传递给函数时才起作用(而不是将列表作为第一个参数传递,而是*导致列表中的每个元素作为一个单独的参数传递,一个接一个地传递) ,就好像在参数列表中用逗号隔开了它们一样)
Cameron

72

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

>>> 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)

举例说明。+1!
AruniRC

28

只是解释概念/使用它的另一种方法。

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]

3
这很重要,在其他地方都没有提到
Gershom

1
这个答案并不特别适用于该问题,而是星号的重要应用(因此我认为在“有雾”的标题下比较合适)。同样,函数定义中还有另一个重要的应用程序:def func(a, b, *args):有关更多信息,请参见此答案
ASL
By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.