9 
import itertools
def _yield_sample():
    it = iter(itertools.combinations('ABCD', 2))
    it2 = iter(itertools.combinations('EFGH', 3))
    itc = itertools.chain(it,it2)
    for x in itc:
        yield x

def main():
    for x in _yield_sample():
        print x

这可以打印组合。

>>> 
('A', 'B')
('A', 'C')
('A', 'D')
...

但是这个：

def __position_combination(_count = [2,3,4,5]):
    its = []
    for ct in _count:
        it = iter(itertools.combinations('ABCDEFG', ct))
        its.append(it)
    itc = itertools.chain(its)
    for x in itc:
        yield x

def main():
    for x in __position_combination():
        print x

不会，它会打印出来

>>> 
<itertools.combinations object at 0x02179210>
<itertools.combinations object at 0x02179240>
<itertools.combinations object at 0x02179270>

我必须重写代码以使其按要求工作。

itc = itertools.chain(*its)

这是我的问题：

itertools.chain(iter, iter, iter)  vs    itertools.chain(*[iter,iter,iter])

他们之间有什么不同？根据文档 ，itertools.chain确实：

def chain(*iterables):
    # chain('ABC', 'DEF') --> A B C D E F
    for it in iterables:
        for element in it:
            yield element

那么，为什么itertools.chain(iter, iter, iter)也可以在这里工作？
是什么意思iter, iter, iter = *(iter, iter, iter)？

4

2 回答 2

16

是的。

但是itertools.chain.from_iterable()。

于 2011-03-08T06:40:41.303 回答
4

它的意思是 iter, iter, iter = *(iter, iter, iter) 吗？

不是自己。参数解包仅在调用函数时有效，因此以下两个调用是等价的：

f(1,2,3)
f(*[1,2,3])

但*[1,2,3]除非它是调用的一部分，否则不会编译。您可以将普通（位置）参数与未打包的参数混合：

f(1,*[2,3])

所以上面也等价于前面的两个调用。

于 2011-03-08T08:48:04.957 回答