python - 如何从列表末尾删除 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]

还有itertools.dropwhile ，但你必须执行两次反转:

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