python - 用最左边的非无值替换列表中的无

Question

给定

a = [None,1,2,3,None,4,None,None]

我愿意

a = [None,1,2,3,3,4,4,4]

目前我用暴力破解它:

def replaceNoneWithLeftmost(val):
    last = None
    ret = []
    for x in val:
        if x is not None:
            ret.append(x)
            last = x
        else:
           ret.append(last)
    return ret

最后，我想去

a = [1,1,2,3,3,4,4,4]

从右向左运行。目前我有

def replaceNoneWithRightmost(val):
    return replaceNoneWithLeftmost(val[::-1])[::-1]

我对就地或创建新列表并不挑剔，但现在这对我来说很奇怪。我看不到存储临时“最后”值和使用 map/lambda 的方法，并且没有想到其他任何东西。

Answer 1

IIUC，您可以使用 itertools.accumulate 生成前向填充:

>>> from itertools import accumulate
>>> a = [None,1,2,3,None,4,None,None]
>>> list(accumulate(a, lambda x,y: y if y is not None else x))
[None, 1, 2, 3, 3, 4, 4, 4]