lemmatize.py

词型还原

import nltk
nltk.download('omw-1.4')
nltk.download('punkt')

from nltk.stem import PorterStemmer
from nltk.stem import LancasterStemmer
from nltk.stem import WordNetLemmatizer
wordnet_lemmatizer = WordNetLemmatizer()
import os

words = []
porter = PorterStemmer()
lancaster = LancasterStemmer()

new_words = [wordnet_lemmatizer.lemmatize(word, pos="v") for word in words]
new_words.sort()
filename = 'words.txt'
with open(os.path.join(os.getcwd(), filename), 'w') as file:
    file.writelines([line + '\n' for line in new_words])