python - 匿名列表如何在 python 列表理解中工作？

Question

给定函数:

def merge(nums1, nums2):
    return [max(nums1, nums2).pop(0) for _ in nums1 + nums2]

我认为它是这样工作的:

def merge(nums1, nums2):
    ans = []
    while nums1 or nums2:
        if nums1 > nums2:
            ans += nums1[0],
            nums1 = nums1[1:]
        else:
            ans += nums2[0],
            nums2 = nums2[1:]
    return ans

该函数取 nums1 和 num2 中当前最左边值的最大值，并将其附加到一个新数组。

谁能给我解释一下为什么第一个等同于第二个？具体来说，部分:

for _ in nums1 + nums2

最让我困惑的是。

类似的比如初始化一个大小为n的数组:

[0 for _ in xrange(n)]

我的理解是

for _ in xrange(n) 

只是告诉执行n次，并用for之前的表达式的返回值填充数组，类似地，

for _ in nums1 + nums2

执行 len(nums1)+len(nums2) 次，但如果我们弹出，长度应该改变。

那我为什么误会了呢？ in子句后面的部分是不是原件的副本？

Answer 1

问题 1:

Can anyone explain to me why the first one is equivalent to the second one?

A1:

我们可以使用这个函数并观察一些东西:

>>> a = [1,2,3]
>>> b = [4,0,5]
>>>
>>> print merge(a,b)
[4, 1, 2, 3, 0, 5]
>>> a
[]
>>> b
[]

max(nums1, nums2).pop(0) 用于比较两个列表的第一个元素。每当它执行一次循环时，它都会弹出最大列表的第一个元素。实际上，max 返回的对象正在使用内存引用，因此这两个列表将逐渐减少它们的元素以完成 for 循环。因此，这就是为什么 a 和 b 变量在函数完成后为空的原因。

让我来展示一下这个函数是如何工作的。

>>> a = [1,2,3]
>>> b = [4,0,5]
>>> merge1(a,b)
Step1
nums1:  [1, 2, 3]
nums2:  [4, 0, 5]
The max list is  [4, 0, 5]
So it will pop : 4
Step2
nums1:  [1, 2, 3]
nums2:  [0, 5]
The max list is  [1, 2, 3]
So it will pop : 1
Step3
nums1:  [2, 3]
nums2:  [0, 5]
The max list is  [2, 3]
So it will pop : 2
Step4
nums1:  [3]
nums2:  [0, 5]
The max list is  [3]
So it will pop : 3
Step5
nums1:  []
nums2:  [0, 5]
The max list is  [0, 5]
So it will pop : 0
Step6
nums1:  []
nums2:  [5]
The max list is  [5]
So it will pop : 5
[4, 1, 2, 3, 0, 5]

然后附上我的代码:

def merge1(nums1, nums2):
    list_ = []
    count = 1
    for _ in nums1 + nums2:
        print 'Step%s' % count
        print 'nums1: ' , nums1
        print 'nums2: ' , nums2
        print 'The max list is ', max(nums1, nums2)
        print 'So it will pop :', max(nums1, nums2)[0]
        list_.append(max(nums1, nums2).pop(0))
        count +=1
    return list_

根据工作流程，你会看到max(nums1, nums2).pop(0)等价于ans += nums1[0];nums1 = nums1[1:] ; 或 ans += nums2[0];nums2 = nums1[1:];。

---

问题 2:

for _ in nums1 + nums2 execute len(nums1)+len(nums2) times, but if we pop, the length should changed.

A2:

不可以，for循环读取数字或对象作为执行计数后，pop不能动态影响执行计数。

我可以为您提供等效的代码。

def merge1(nums1, nums2):
    return [max(nums1, nums2).pop(0) for _ in range(len(nums1) + len(nums2)]

演示:

>>> a = [1,2,3]
>>> b = [4,0,5]
>>> print merge1(a,b)
[4, 1, 2, 3, 0, 5]

那么它发生了吗？什么都没有。

---

问题 3:

Specifically, the part: for _ in nums1 + nums2

A3:

根据我的A2，你可以把它想象成for _ in range(len(nums1) + len(nums2)，它们是等价的。for循环只需要元素的数量作为执行计数。