使用itertools.islice时重复首尾元素？遍历列表连续元素并获取索引

嘿，我明白你的需求了——你想要遍历列表时，同时获取连续两个元素的索引和值，并且要让最后一个元素和第一个元素配对（也就是首尾循环衔接），对吧？

先看看你原来的代码问题：它用zip(enumerate(my_list), itertools.islice(enumerate(my_list), 1, None))只能处理到原列表的倒数第二个元素，最后一个元素没有和第一个元素配对。下面给你两种可行的解决方案：

方法一：扩展列表实现首尾衔接 #

最直观的方式是把原列表的第一个元素追加到列表末尾，这样遍历的时候就能自然衔接首尾。然后控制遍历次数和原列表长度一致，避免多余的配对：

import itertools

my_list = ['A','C','G','T','T','C','G','A','G']
# 把第一个元素加到列表末尾，实现首尾循环的基础
extended_list = my_list + [my_list[0]]

print("原列表:", my_list)
print()

# 遍历原列表的长度次数，刚好覆盖所有连续配对（含最后一个与第一个）
for (n1, v1), (n2, v2) in zip(enumerate(my_list), itertools.islice(enumerate(extended_list), 1, None)):
    print(f"索引{n1}: {v1}  →  索引{n2}: {v2}")
    print()

代码说明： #

• extended_list让原列表的最后一个元素之后紧跟第一个元素，确保islice能取到衔接的元素
• enumerate(my_list)保证我们只获取原列表的索引，而islice(enumerate(extended_list), 1, None)会从索引1开始取到最后，刚好和原列表的索引一一对应，形成9对连续元素（包括最后一对G和A）

方法二：用itertools.cycle实现循环遍历 #

如果你不想修改原列表，可以用itertools.cycle创建一个循环迭代器，手动控制遍历次数来实现首尾衔接：

import itertools

my_list = ['A','C','G','T','T','C','G','A','G']
# 创建带索引的循环迭代器
cycled_pairs = itertools.cycle(enumerate(my_list))
# 先取出第一个元素作为初始的"前一个元素"
prev_idx, prev_val = next(cycled_pairs)

# 遍历原列表长度次，每次取循环迭代器的下一个元素
for _ in range(len(my_list)):
    curr_idx, curr_val = next(cycled_pairs)
    print(f"索引{prev_idx}: {prev_val}  →  索引{curr_idx}: {curr_val}")
    print()
    # 更新前一个元素为当前元素，用于下一次循环
    prev_idx, prev_val = curr_idx, curr_val

代码说明： #

• itertools.cycle会无限循环遍历带索引的原列表，所以我们用range(len(my_list))控制只遍历9次
• 先取出第一个元素作为起始，之后每次取循环迭代器的下一个元素，最后一次会取到原列表的第一个元素，完美实现首尾衔接

两种方法都能实现你要的效果，你可以根据自己的喜好选择~

内容的提问来源于stack exchange，提问作者everestial