如何从Python的特定点开始Itertools循环？

Question

我已经使用以下代码为英语字母创建了一个itertools循环，

lowercase_letters_cycle = itertools.cycle(string.ascii_lowercase)

如果我在此迭代对象上运行循环 loop，则第一个迭代会给我“ a”输出是因为周期从“ A”开始。我该如何做到这一点，以便循环从我选择的任何字母开始？

有效的一种方法是，

def start_cycle(letter):
  lowercase_letters_cycle = itertools.cycle(lowercase_letters)
  letter_index = lowercase_letters.index(letter)
  index = 0

  while True:
    if index == letter_index:
      break

    letter = next(lowercase_letters_cycle)
    index += 1

  return lowercase_letters_cycle

但是有任何较短的方法吗？

Answer 1

itertools文档提供A 用于从迭代器中食用许多项目。

from itertools import islice
import collections


def consume(iterator, n=None):
    "Advance the iterator n-steps ahead. If n is None, consume entirely."
    # Use functions that consume iterators at C speed.
    if n is None:
        # feed the entire iterator into a zero-length deque
        collections.deque(iterator, maxlen=0)
    else:
        # advance to the empty slice starting at position n
        next(islice(iterator, n, n), None)

因此，您可以创建周期，然后在继续之前消耗一些周期。

lowercase_letters_cycle = itertools.cycle(string.ascii_lowercase)

consume(lowercase_letters_cycle, ord('n') - ord('a'))

assert next(lowercase_letters_cycle) == 'n')

相同的消费也可以从第三方更多 - itertools软件包。

Answer 2

您可以从itertools模块中组合ISLICE cyce cyce ，如下所示：

import string 
import itertools

my_it = itertools.islice(itertools.cycle(string.ascii_lowercase), 3, None)

它将产生d（来源的字符在第一个3）之后，然后e，....，然后z，然后a，然后b ， 等等。您可以将第二个参数中的数字更改为islice以从不同的字母开始。

Answer 3

您可以将输入列表切成小块，以从您选择的索引开始。

offset = 10
lowercase_letters = list(string.ascii_lowercase)

#                 offset_index to end       + start to offset_index-1
offset_letters = lowercase_letters[offset:] + lowercase_letters[:offset]
offset_letters_cycle = itertools.cycle(offset_letters)

然后，

for i in range(10):
    print(next(offset_letters_cycle), end=" ")

打印：

k l m n o p q r s t 

Answer 4

import string
from itertools import cycle, dropwhile

def start_cycle(letter):
    return dropwhile(lambda x: x!= letter, cycle(string.ascii_lowercase))

itertools.drophiles.dropwhiles.dropwhile 其前哨功能的效果结果 - 在这种情况下，lambda x：x！= letter，并将“吞咽”结果，直到函数首次返回false。那时
该功能不再被调用，并且可以进行，从而产生任何
进一步的值。

另外，请注意，无需将ascii_lowercase转换为列表，
因为字符串已经是迭代的。

请记住，如果通过ASCII_LOWERCASE中的字符，这将陷入100％CPU的无限循环中。最好用一张检查：

def start_cycle(letter):
    if letter not in string.ascii_lowercase:
         raise ValueError()
    return dropwhile(lambda x: x!= letter, cycle(string.ascii_lowercase))

Answer 5

如果您只想从特定点开始浏览列表，则可以这样做：

my_list = ['a', 'b', 'c', 'd', 'e', 'f', 'g']
start_index = 4

for i in my_list[start_index:]+my_list[:start_index]:
    print(i)