使用两个列表的嵌套列表推导式

Question

我理解如何使用简单的列表推导式，例如：

[x*2 for x in range(5)] # returns [0,2,4,6,8]

而且我也理解嵌套的列表推导式是如何工作的：

w_list = ["i_have_a_doubt", "with_the","nested_lists_comprehensions"]

# returns the list of strings without underscore and capitalized
print [replaced.title() for replaced in [el.replace("_"," ")for el in w_list]]

所以，当我试图做这件事时

l1 = [100,200,300]
l2 = [0,1,2]
[x + y for x in l2 for y in l1 ]

我原本期待这个:

[100,201,302]

但是我得到了这个：

[100,200,300,101,201,301,102,202,302]

所以我找到了一个更好的解决问题的方法，它给了我想要的结果。

[x + y for x,y in zip(l1,l2)]

但我不理解第一段代码返回9个元素的原因。

- André Caldas

Answer 1

它有9个数字的原因是因为Python会将

[x + y for x in l2 for y in l1 ]

类似于

for x in l2:
    for y in l1:
       x + y

即，它是一个嵌套循环。

- rspencer

Answer 2

列表推导式相当于 for 循环。因此，[x + y for x in l2 for y in l1 ] 将变成：

new_list = []
for x in l2:
    for y in l1:
        new_list.append(x + y)

zip 函数返回包含每个列表中一个元素的元组。因此，[x + y for x,y in zip(l1,l2)] 相当于：

new_list = []
assert len(l1) == len(l2)
for index in xrange(len(l1)):
    new_list.append(l1[index] + l2[index])

- Wesley Baugh

Answer 3

上述答案已经足够回答你的问题，但我想提供一个列表理解的解决方案供你参考（因为那是你最初提供的代码和你想要理解的内容）。

假设两个列表的长度相同，你可以这样做：

[l1[i] + l2[i] for i in range(0, len(l1))]

- Sticky

---

或者，您可以枚举其中一个列表，这样您只需要索引其中一个即可。 - Mad Physicist

xrange更好的内存优化。 - Mickey Perlstein

@MickeyPerlstein 这有点晚了，但在Python 3中，range等同于xrange。 - sohnryang

Answer 4

[x + y for x in l2 for y in l1 ]

等同于：

lis = []
for x in l:
   for y in l1:
      lis.append(x+y)

对于你正在迭代的列表l中的每个元素，你都会再次迭代l2，因为l有3个元素，而l1有元素，所以总循环次数等于9 (len(l)*len(l1))。

- Ashwini Chaudhary

Answer 5

这个序列

res = [x + y for x in l2 for y in l1 ]

等价于

res =[]
for x in l2:
    for y in l1:
        res.append(x+y)

- oleg