python 将列表中的元素完成特定的向右移动,参数:列表、移动长度 - 如:[1, 2, 3, 4, 5],移动2,结果:4, 5, 1, 2, 3

python
def right_shift(lt, step):
    step %= len(lt)
    '''
    for i in range(step):
        # 弹出最后一个元素
        last = lt.pop()
        # 将最后一个元素插入到开头
        lt.insert(0, last)
    '''
    left = lt[:-step]
    right = lt[-step:]
    lt.clear()
    lt.extend(right)
    lt.extend(left)


lt = [1, 2, 3, 4, 5]
right_shift(lt, 2)
print(lt)

猜你喜欢

转载自blog.csdn.net/qq_42795565/article/details/81317648