python - 如何对句子列表进行词形还原

Question

如何在 Python 中对句子列表进行词形还原？

from nltk.stem.wordnet import WordNetLemmatizer
a = ['i like cars', 'cats are the best']
lmtzr = WordNetLemmatizer()
lemmatized = [lmtzr.lemmatize(word) for word in a]
print(lemmatized)

这是我尝试过的方法，但它给了我相同的句子。我需要先对单词进行分词才能正常工作吗？

Answer 1

长话短说:

pip3 install -U pywsd

然后:

>>> from pywsd.utils import lemmatize_sentence

>>> text = 'i like cars'
>>> lemmatize_sentence(text)
['i', 'like', 'car']
>>> lemmatize_sentence(text, keepWordPOS=True)
(['i', 'like', 'cars'], ['i', 'like', 'car'], ['n', 'v', 'n'])

>>> text = 'The cat likes cars'
>>> lemmatize_sentence(text, keepWordPOS=True)
(['The', 'cat', 'likes', 'cars'], ['the', 'cat', 'like', 'car'], [None, 'n', 'v', 'n'])

>>> text = 'The lazy brown fox jumps, and the cat likes cars.'
>>> lemmatize_sentence(text)
['the', 'lazy', 'brown', 'fox', 'jump', ',', 'and', 'the', 'cat', 'like', 'car', '.']

---

否则，看看pywsd中的函数是怎样的:

• 标记字符串
• 使用词性标注器并映射到 WordNet 词性标签集
• 试图阻止
• 最后用 POS 和/或词干调用词形还原器

参见 https://github.com/alvations/pywsd/blob/master/pywsd/utils.py#L129