如何将带有比较的 for 循环转换为列表理解？

如何解决如何将带有比较的 for 循环转换为列表理解？

我正在努力提高使用列表理解的能力，这段代码适用于我将其用于（代码挑战）的目的，但我想使用列表理解并理解其工作原理。

j = 0
for i in range(len(a)):
    if (a[i] >= 0):
        a[i] = ans[j]
        j += 1

我试过了

lst = [x for (i,x) in enumerate(a) if x >= 0]

但显然我在这里误解了一些东西，因为它不起作用。

解决方法

这是您的代码以及算法的解释：

j = 0                    # set the "pointer" on `ans` to the start of `ans`
for i in range(len(a)):  # for every element `e` of `a`
    if (a[i] >= 0):      # if `e` is positive
        a[i] = ans[j]    # replace it by the current element of `ans`
        j += 1           # and advance the "pointer" on `ans`.

基本上，您迭代 a，当您找到一个正数时，将其替换为 ans 的下一个元素。让我们用一个例子试试：

a = [-1,2,-3,5,7,-9]
ans = [10,20,30,40]
 
...
         
print(a)
# [-1,10,-9]

如果您想使用列表推导式，您必须为 a 的每个元素创建包含 ans 的下一个元素（如果元素为正）或元素本身的列表.然后将新列表分配给 a。很简单：

a = [ans.pop(0) if e >= 0 else e for e in a]
print(a)
# [-1,-9]

显然，如果您不想消耗 ans，您可以使用：

temp = list(ans)
a = [e if e<0 else temp.pop(0) for e in a]

,

由于您有条件地通过 ans，您需要使用迭代器或等效的遍历机制来避免在过程中破坏 ans（如 ans.pop(0) 那样） :

ans = [-2,-4,-7]
a   = [-1,1,-5,-6,3,-8,-9]

a   = [ n if n<0 else next(ans_j) for ans_j in [iter(ans)] for n in a]

print(a)
[-1,-2,-7,-9]