лемматизировать слова в вложенном списке

Как лемматизировать слова во вложенном списке в одну строку? Я пробовал несколько вещей, я приближаюсь, но я думаю, что могу ошибаться в синтаксисе? Как это исправить?

from nltk.stem import WordNetLemmatizer 
lemmatizer = WordNetLemmatizer()
word_list = [['test','exams','projects'],['math','exam','things']]
word_list # type list

Try #1: Does the lemmatization but in different format
for word in word_list:
    for e in word: 
        print(lemmatizer.lemmatize(e)) # not the result I need for 

Try #2: Looking for similar approach in one line to solve the problem. Not giving correct results. 
[[word for word in lemmatizer.lemmatize(str(doc))] for doc in word_list]

Output needed: 

[['test','exam','project'],['math','exam','thing']]

python-3.x nltk lemmatization
person sharp    schedule 13.01.2021    source источник

Ответы (1)


arrow_upward
0
arrow_downward

Я нашел решение цикла for для моего вопроса выше. Я не мог получить это в одну строку, но пока это работает. Если кто-то ищет решение:

word_list_lemma = []
for ls in word_list:
    word_lem = []
    for word in ls:
        word_lem.append(lemmatizer.lemmatize(word))
    word_list_lemma.append(word_lem)
person sharp    schedule 14.01.2021