嵌套列表理解有两个列表

时间:2013-05-15 14:39:49

标签: python nested list-comprehension

我理解简单列表理解是如何工作的,例如:

[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 ]

我期待这个:

[101,202,303]

但我得到了这个:

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

所以我有一个更好的方法解决问题,这给了我想要的东西

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

但是我不理解第一个代码上的9个元素的返回

5 个答案:

答案 0 :(得分:48)

它有9个数字的原因是因为python处理

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

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

即,它是一个嵌套循环

答案 1 :(得分:16)

列表推导等同于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])

答案 2 :(得分:9)

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

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

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

答案 3 :(得分: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({ {1}})。

答案 4 :(得分:1)

这个序列

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

相当于

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