旋转列表元素
将指定数量的元素移动到列表的开头。
- 使用切片表示法获取列表的两个切片,并在返回之前将它们组合起来。
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]
将指定数量的元素移动到列表的开头。
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]