lemmatization

lemma.py

import nltk
from nltk.stem import WordNetLemmatizer
# Instantiating the lemmatizer class from nltk
wordnet_lemmatizer = WordNetLemmatizer()
# List of verbs and adjectives to feed to the lemmatizer
verbs = ["thinking", "wondering", "reflecting"]
adjectives = ["better", "brighter", "smarter"]
# Making a nice print for both
print("Verbs")
print("{0:20}{1:20}".format("Word","Lemma"))
for verb in verbs:
 print ("{0:20}{1:20}".format(verb,wordnet_lemmatizer.lemmatize(verb, pos="v")))
print("\n")
print("Adjectives")
print("{0:20}{1:20}".format("Word","Lemma"))
for adj in adjectives:
 print ("{0:20}{1:20}".format(adj,wordnet_lemmatizer.lemmatize(adj, pos="a")))