python - 检查单词是否在 NLTK 同义词集中的最快方法？

Question

我想检查 NLTK 的同义词集中是否存在某些词。下面的代码就是这样做的，

from nltk.corpus import wordnet
if wordnet.synsets(word): 
    ... do something ...

但是如果你有很多词要检查，它会很慢。有没有更快的方法？ 我不需要实际的 synset 对象，只要是/否(如果有的话)。我没有提前的单词列表，所以我无法预先计算答案。

Answer 1

因为您只需要知道哪些单词匹配，所以形成一组所有引理并在那里查找您的单词。形成集合的速度非常快(当然集合查找甚至更快)。

wn_lemmas = set(wordnet.all_lemma_names())
...
if word in wn_lemmas:
    <do something>