动态长度的python生成器

Question

我有这个：

def get_set(line, n=3):
    words = line.split()
    for i in range(len(words) - n):
        yield (words[i], words[i+1], words[i+2])

for i in get_set('This is a test'):
    print(i)

但是，正如您在yield调用中看到的那样，它经过硬编码才能与3一起工作。如何重写yield行以与通过n kwarg传递的任何数字一起工作？

（代码生成器在一个句子中每三个连续单词的集合，希望它生成我作为n传递的所有内容）

Answer 1

您总是可以将元组超出范围

def get_set(line, n=3):
    words = line.split()
    for i in range(len(words) - (n-1)):
        yield tuple(words[i:i+n])

请注意，您需要在len(words) - (n-1)而不是len(words)-n len(words) - (n-1)范围内进行迭代，以获取所有连续对。

用

for i in get_set('This is a very long test'):
    print(i)

这给出：

n = 3：

('This', 'is', 'a') ('is', 'a', 'very') ('a', 'very', 'long') ('very', 'long', 'test')

n = 4：

('This', 'is', 'a', 'very') ('is', 'a', 'very', 'long') ('a', 'very', 'long', 'test')

Answer 2

for row in zip(*[words[x:] for x in range(n)]):
    yield row

我认为应该工作

 for i in range(len(words)-n):
     yield words[i:i+n]

也应该工作...

（如果需要，可以投射到元组中……）

Answer 3

您可以在列表上使用切片 ：

def get_set(line, n=3):
    words = line.split()
    for i in range(0, len(words), n):
        yield words[i:i+n]

for i in get_set('This is a test'):
    print(i)

['This', 'is', 'a']
['test']

for i in get_set('This is another very boring test', n=2):
    print(i)

['This', 'is']
['another', 'very']
['boring', 'test']