如何从列表末尾删除 None 的所有实例？

Question

Python 有一个名为 rstrip() 的字符串方法：

>>> s = "hello world!!!"
>>> s.rstrip("!")
'hello world'

我想为 Python 列表实现类似的功能。也就是说，我想从列表末尾删除给定值的所有实例。在本例中，该值为None。

这是一些开始的例子：

[1, 2, 3, None]
[1, 2, 3, None, None, None]
[1, 2, 3, None, 4, 5]
[1, 2, 3, None, None, 4, 5, None, None]

我希望最终结果是：

[1, 2, 3]
[1, 2, 3]
[1, 2, 3, None, 4, 5]
[1, 2, 3, None, None, 4, 5]

这是到目前为止我的解决方案：

while l[-1] is None:
    l.pop()

Answer 1

如果您想就地修改列表，那么您的解决方案很好，只需确保处理列表为空时的情况：

while l and l[-1] is None:
    l.pop()

如果您想计算一个新列表，您可以将您的解决方案调整为：

def stripNone(l):
    if not l:
        return []
    
    rlim = 0
    for x in reversed(l):
        if x is None:
            rlim += 1
        else:
            break
    
    return l[: len(l) - rlim]

还有 < a href="https://docs.python.org/3/library/itertools.html#itertools.dropwhile" rel="nofollow noreferrer">itertools.dropwhile，但是你必须执行两个逆转：

def stripNone(l):
    return list(dropwhile(lambda x: x is None, l[::-1]))[::-1]

Answer 2

另外两个版本也适用于仅 None 的列表：

while None in l[-1:]:
    l.pop()

for x in reversed(l):
    if x is not None:
        break
    l.pop()

对 l = [None] * 10**3 上的一些解决方案进行基准测试：

 83 us  stripNone1
137 us  stripNone2
 60 us  stripNone3
 42 us  stripNone3b
 53 us  stripNone4
 34 us  stripNone5
 19 us  stripNone6

请注意 stripNone2 和 stripNone6 有一个小缺陷：如果 Trailline None 中有一个对象不是 None 但声称​​等于 None，那么它将被删除。不过，这样的物体非常不寻常。也许人们实际上也想要删除这样一个对象。

基准代码：

def stripNone1(l):
    while l and l[-1] is None:
        l.pop()

def stripNone2(l):
    while None in l[-1:]:
        l.pop()

def stripNone3(l):
    for x in reversed(l):
        if x is not None:
            break
        l.pop()

def stripNone3b(l):
    pop = l.pop
    for x in reversed(l):
        if x is not None:
            break
        pop()

def stripNone4(l):
    for i, x in enumerate(reversed(l), 1):
        if x is not None:
            del l[-i:]
            break

def stripNone5(l):
    pop = l.pop
    try:
        while (last := pop()) is None:
            pass
        l.append(last)
    except IndexError:
        pass

def stripNone6(l):
    while l:
        chunk = l[-32:]
        if chunk.count(None) < len(chunk):
            while l[-1] is None:
                l.pop()
            break
        del l[-32:]

from timeit import repeat
solutions = stripNone1, stripNone2, stripNone3, stripNone3b, stripNone4, stripNone5, stripNone6
for i in range(3):
    print(f'Round {i+1}:')
    for sol in solutions:
        ls = [[42] * head + [None] * tail
              for _ in range(5)
              for head in range(0, 2001, 200)
              for tail in range(0, 2001, 200)]
        number = len(ls) // 5
        ls = iter(ls)
        time = min(repeat(lambda: sol(next(ls)), number=number)) / number
        print(f'{int(time * 10**6):3d} us  {sol.__name__}')