具有多個變量的列表理解

Question

有什麼方法可以使列表理解中的兩個或多個項目出列。

I.e.我想要做這樣的事情：

>>> q = [1, 2, 3, 4] 
>>> r = [x+y for x, y in q] 
** This does not work though :(** 

預計：

>>> r 
[3, 7] 

Answer 1

的建議您可以從itertools文檔中使用的石斑魚recipe：

>>> from itertools import izip_longest 
>>> [x + y for x, y in grouper(2,q)] 
[3, 7] 

的源代碼爲配方是這樣的：

def grouper(n, iterable, fillvalue=None): 
    "grouper(3, 'ABCDEFG', 'x') --> ABC DEF Gxx" 
    args = [iter(iterable)] * n 
    return izip_longest(fillvalue=fillvalue, *args) 

Answer 2

作品，如果列表中只有2如組。爲[1,2,3,4,5]不會工作，但會爲[1,2,3,4,5,6]

>>> q = [1, 2, 3, 4] 
>>> [x+y for x, y in zip(*[iter(q)]*2)] 
[3, 7] 

也通過工作@gnibbler

>>> map(sum, zip(*[iter(q)]*2)) 

Answer 3

另一種解決方案

>>> [sum(q[i:i+2]) for i in range(0,len(q),2)] 
[3, 7] 
>>> q=[1,2,3,4,5] 
>>> [sum(q[i:i+2]) for i in range(0,len(q),2)] 
[3, 7, 5]