Quick concordance search for a corpus.

concordance.py

#!/usr/bin/env python

from nltk import Text
from nltk.corpus import PlaintextCorpusReader


CORPUS_ROOT = '/Users/benjamin/Development/corpora/gutenburg/'

if __name__ == "__main__":
    reader = PlaintextCorpusReader(CORPUS_ROOT, '.*')
    text   = Text(reader.words())
    while True:
        word = raw_input('Enter word: ')
        print (word);
        text.concordance(word)