67

我了解简单列表理解的工作原理,例如:

[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 个元素的返回

4

5 回答 5

86

它有 9 个数字的原因是因为 python 对待

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

类似于

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

即,它是一个嵌套循环

于 2013-05-15T14:44:01.533 回答
29

列表推导等同于 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])
于 2013-05-15T14:45:31.230 回答
21

以上答案足以解决您的问题,但我想为您提供一个列表理解解决方案以供参考(因为那是您的初始代码以及您想要理解的内容)。

假设两个列表的长度相同,您可以这样做:

[l1[i] + l2[i] for i in range(0, len(l1))]
于 2016-06-14T04:35:26.063 回答
3
[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))。

于 2013-05-15T14:44:09.917 回答
2

这个序列

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

相当于

res =[]
for x in l2:
    for y in l1:
        res.append(x+y)
于 2013-05-15T14:47:41.643 回答