Nested list comprehension with two lists

后端 未结 5 891
误落风尘
误落风尘 2020-12-08 06:09

I understand how the simple list comprehension works eg.:

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

and also I understand how the nested

相关标签:
5条回答
  • 2020-12-08 06:51
    [x + y for x in l2 for y in l1 ]
    

    is equivalent to :

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

    So for every element of l you're iterating l2 again and again, as l has 3 elements and l1 has elements so total loops equal 9(len(l)*len(l1)).

    0 讨论(0)
  • 2020-12-08 06:58

    List comprehensions are equivalent to for-loops. Therefore, [x + y for x in l2 for y in l1 ] would become:

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

    Whereas zip returns tuples containing one element from each list. Therefore [x + y for x,y in zip(l1,l2)] is equivalent to:

    new_list = []
    assert len(l1) == len(l2)
    for index in xrange(len(l1)):
        new_list.append(l1[index] + l2[index])
    
    0 讨论(0)
  • 2020-12-08 07:03

    this sequence

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

    is equivalent to

    res =[]
    for x in l2:
        for y in l1:
            res.append(x+y)
    
    0 讨论(0)
  • 2020-12-08 07:06

    The reason it has 9 numbers is because python treats

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

    similarly to

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

    ie, it is a nested loop

    0 讨论(0)
  • 2020-12-08 07:09

    The above answers will suffice for your question but I wanted to provide you with a list comprehension solution for reference (seeing as that was your initial code and what you're trying to understand).

    Assuming the length of both lists are the same, you could do:

    [l1[i] + l2[i] for i in range(0, len(l1))]
    
    0 讨论(0)
提交回复
热议问题