python - 提取搜索词周围的词

Question

我有这个脚本可以在文本中进行单词搜索。搜索进行得非常顺利，结果按预期工作。我想要实现的是提取接近匹配项的 n 个词。例如:

The world is a small place, we should try to take care of it.

假设我正在寻找 place 并且我需要提取右边的 3 个词和左边的 3 个词。在这种情况下，它们将是:

left -> [is, a, small]
right -> [we, should, try]

执行此操作的最佳方法是什么？

谢谢!

Answer 1

def search(text,n):
    '''Searches for text, and retrieves n words either side of the text, which are retuned seperatly'''
    word = r"\W*([\w]+)"
    groups = re.search(r'{}\W*{}{}'.format(word*n,'place',word*n), text).groups()
    return groups[:n],groups[n:]

这允许您指定要捕获的任一侧的单词数。它通过动态构建正则表达式来工作。与

t = "The world is a small place, we should try to take care of it."
search(t,3)
(('is', 'a', 'small'), ('we', 'should', 'try'))