python列表中元素移动,Python:以固定长度移动列表中的元素

I'm looking for an elegant way to write a simple function that would shift the elements of list by a given number of positions, while keeping the list of the same length and padding empty positions with a default value. This would be the docstring of the function:

def shift_list(l, shift, empty=0):

"""

Shifts the elements of a list **l** of **shift** positions,

padding new items with **empty**::

>>> l = [0, 1, 4, 5, 7, 0]

>>> shift_list(l, 3)

[0, 0, 0, 0, 1, 4]

>>> shift_list(l, -3)

[5, 7, 0, 0, 0, 0]

>>> shift_list(l, -8)

[0, 0, 0, 0, 0, 0]

"""

pass

How would you proceed ? Any help greatly appreciated !

解决方案

I'd use slice assignment:

def shift_list(l, shift, empty=0):

src_index = max(-shift, 0)

dst_index = max(shift, 0)

length = max(len(l) - abs(shift), 0)

new_l = [empty] * len(l)

new_l[dst_index:dst_index + length] = l[src_index:src_index + length]

return new_l

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值