旋转列表元素

将指定数量的元素移动到列表的开头。

  • 使用切片表示法获取列表的两个切片,并在返回之前将它们组合起来。
def roll(lst, offset):
  return lst[-offset:] + lst[:-offset]

roll([1, 2, 3, 4, 5], 2) # [4, 5, 1, 2, 3]
roll([1, 2, 3, 4, 5], -2) # [3, 4, 5, 1, 2]