Wrapping around a python list as a slice operation

2020-01-25 09:50发布

Consider the following simple python code

>>> L = range(3)
>>> L
[0, 1, 2]

We can take slices of this array as follows:

>>> L[1:3]
[1, 2]

Is there any way to wrap around the above array by shifting to the left

[1, 2, 0]

by simply using slice operations?

标签: python slice
4条回答
▲ chillily
2楼-- · 2020-01-25 10:30

Left:

L[:1], L[1:] = L[-1:], L[:-1]

Right:

L[-1:], L[:-1] = L[:1], L[1:]
查看更多
老娘就宠你
3楼-- · 2020-01-25 10:35

To my mind, there's no way, unless you agree to cut and concatenate lists as shown above. To make the wrapping you describe you need to alter both starting and finishing index.

  • A positive starting index cuts away some of initial items.
  • A negative starting index gives you some of the tail items, cutting initial items again.
  • A positive finishing index cuts away some of the tail items.
  • A negative finishing index gives you some of the initial items, cutting tail items again.

No combination of these can provide the wrapping point where tail items are followed by initial items. So the entire thing can't be created.

Numerous workarounds exist. See answers above, see also itertools.islice and .chain for a no-copy sequential approach if sequential access is what you need (e.g. in a loop).

查看更多
甜甜的少女心
4楼-- · 2020-01-25 10:38

Rotate left n elements (or right for negative n):

L = L[n:] + L[:n]

Note that collections.deque has support for rotations. It might be better to use that instead of lists.

查看更多
【Aperson】
5楼-- · 2020-01-25 10:54

If you are not overly attached to the exact slicing syntax, you can write a function that produces the desired output including the wrapping behavior.

E.g., like this:

def wrapping_slice(lst, *args):
    return [lst[i%len(lst)] for i in range(*args)]

Example output:

>>> L = range(3)
>>> wrapping_slice(L, 1, 4)
[1, 2, 0]
>>> wrapping_slice(L, -1, 4)
[2, 0, 1, 2, 0]
>>> wrapping_slice(L, -1, 4, 2)
[2, 1, 0]

Caveat: You can't use this on the left-hand side of a slice assignment.

查看更多
登录 后发表回答