《Python自然语言处理》第一章 习题解答 练习17

问题描述：

　　使用text9.index()查找词sunset的索引值。你需要将这个词作为一个参数插入到圆括号之间。通过尝试和出错的过程中，找到完整的句子中包含这个词的切片。

解题思路：

　　用两个集合，一个存放sunset的位置，一个存放标点符号的位置，然后将sunset插入到标点符号之间。

代码：

import nltk
from nltk.book import *


def solution(keyword, text):
    # keyword = 'sunset'
    pun = [i for i, j in enumerate(text) if j == '.' or j == '?' or j == '!']
    sunset = [i for i, j in enumerate(text) if j == keyword]
    ans = []
    flag = 0
    for i in range(len(sunset)):
        j = flag
        while j < len(pun):
            if sunset[i] < pun[j] and pun[j - 1] < sunset[i]:
                ans.append([pun[j - 1] + 1, pun[j] + 1])
                flag = j + 1
                break
            j += 1
    # print ans
    for i in range(len(ans)):
        print 'start from %d to end in %d' % (ans[i][0], ans[i][1])
        print ' '.join(text[ans[i][0]: ans[i][1]])

if __name__ == "__main__":
    solution()
    raw_input()