的Python的发电机跃

Question

我在python3问题

def generator(string.ascii_lowercase, [0,1,2,3]): 
    pass 

的解决方案应该是：

'a', 'b', 'd', 'g', 'k' 

你有一个想法，我怎么能实现这个???

我0,1,2,3是跳指数

Answer 1

你可以使用一个iterator（迭代器，因为它简化了提前步骤）和嵌套的循环使用next：

def generator(s, steps): 
    it = iter(s)   # create an iterator over the string 
    yield next(it)   # yield the first item and advance the iterator thereby 
    for i in steps:   # go through the list of steps 
     for _ in range(i): # advance the iterator by "i" steps without returning 
      next(it) 
     yield next(it)  # yield again, after the iterator has been advanced. 

这给预期的结果：

>>> import string 
>>> list(generator(string.ascii_lowercase, [0,1,2,3])) 
['a', 'b', 'd', 'g', 'k'] 

---

随着itertools.islice你可以简化和推进下一个项目的收益：

import itertools 

def generator(s, steps): 
    it = iter(s) 
    yield next(it) 
    for i in steps: 
     yield next(itertools.islice(it, i, None)) 

如果你不喜欢的迭代器您可以随时使用正常的索引（假设你有一个可以索引的顺序 - 一个字符串作品但发电机不会）：

def generator(s, steps): 
    yield s[0] 
    idx = 1 
    for i in steps: 
     idx += i 
     yield s[idx] 
     idx += 1