Python 列表理解 : Modify list elements if a certain value occurs

Question

如何在 Python 的列表理解中执行以下操作？

nums = [1,1,0,1,1]
oFlag = 1
res = []
for x in nums:
    if x == 0:
        oFlag = 0
    res.append(oFlag)
print(res)

# Output: [1,1,0,0,0]

基本上在这个例子中，一旦 0 出现，就将列表的其余部分归零。

Answer 1

在某些情况下，列表推导式是一种针对 map 和 filter 函数的“命令式”语法，存在于许多函数式编程语言中。您尝试执行的操作通常称为 accumulate，这是一个略有不同的操作。您不能根据 map 和 filter 实现 accumulate，除非使用副作用。 Python 允许您在列表推导式中产生副作用，因此这绝对是可能的，但带有副作用的列表推导式有点靠不住。以下是使用 accumulate 实现此功能的方法:

nums = [1,1,0,1,1]

def accumulator(last, cur):
    return 1 if (last == 1 and cur == 1) else 0

list(accumulate(nums, accumulator))

或在一行中:

list(accumulate(nums, lambda last, cur: 1 if (last == 1 and cur == 1) else 0))

当然，有几种方法可以使用外部状态和带有副作用的列表推导来做到这一点。这是一个示例，它有点冗长但非常明确地说明了状态是如何被操纵的:

class MyState:
    def __init__(self, initial_state):
        self.state = initial_state
    def getNext(self, cur):
        self.state = accumulator(self.state, cur)
        return self.state

mystate = MyState(1)
[mystate.getNext(x) for x in nums]