从列表中返回子列表，其中最后一个元素链接回列表中的起始索引

Question

如何从返回 3 个连续元素的列表中返回子列表，其中最后一个元素链接回第一个给定列表中的任何索引？

例如，给定指标3的list = [1,2,3,4,5]，将返回[4,5,1]。或者给定4带有 list = 的索引[1,2,3,4,5]，将返回[5,1,2].

我有的选项：

1. return list[index:] + list[:index+3]

2. return list[index:index+3] + list[:len(list)-index]

3. return list[index+3:] + list[:len(list)-index]

4. return list[index:index+3] + list[:max(0 , -1*(len(list)-index-3))]

Answer 1

模运算符的%典型用例：lst = [1,2,3,4,5]  # do not shadow built-in 'list'i = 3[lst[x % len(lst)] for x in range(i, i+3)]# [4, 5, 1]i = 4[lst[x % len(lst)] for x in range(i, i+3)]# [5, 1, 2]从您给定的选项中，最后一个 (4.) 是产生相同结果的选项：lst[i:i+3] + lst[:max(0 , -1*(len(lst)-i-3))]您只需尝试一下即可轻松确认;)

Answer 2

这可以使用np.roll轻松实现：lst = [1,2,3,4,5]def roll_n(x, index, length=3):    return np.roll(x,-index)[:length].tolist()roll_n(lst,3)[4, 5, 1]roll_n(lst,4)[5, 1, 2]