python获取列表的前x个元素并删除它们

Question

注意:我知道 StackOverflow 上可能已经有这个问题的答案，但我找不到。

我需要这样做:

>>> lst = [1, 2, 3, 4, 5, 6]
>>> first_two = lst.magic_pop(2)
>>> first_two
[1, 2]
>>> lst
[3, 4, 5, 6]

现在 magic_pop 不存在，我用它只是为了展示我需要的例子。是否有像 magic_pop 这样的方法可以帮助我以 Pythonic 的方式做所有事情？

Answer 1

分两步进行。使用切片获取前两个元素，然后从列表中删除该切片。

first_list = lst[:2]
del lst[:2]

如果你想要一个单行代码，你可以将它包装在一个函数中。

def splice(lst, start = 0, end = None):
    if end is None:
        end = len(lst)
    partial = lst[start:end]
    del lst[start:end]
    return partial

first_list = splice(lst, end = 2)