移除列表元素

从列表的开头移除 n 个元素,并返回一个新的列表。

  • 使用切片操作符从列表的开头创建一个包含 n 个元素的切片。
def take(itr, n = 1):
  return itr[:n]

take([1, 2, 3], 5) # [1, 2, 3]
take([1, 2, 3], 0) # []