python - 如何将词干字符串存储到单个数组中？

Question

我是 python 新手，我只是想编写一个简单的程序，将词干在一个字符串中，并输出一个包含词干的数组/列表，但我似乎可以将它们放入一个数组中。这是我拥有的代码，我包含了输出。提前感谢您的帮助！

from nltk.stem import PorterStemmer
from nltk.tokenize import sent_tokenize, word_tokenize

ps = PorterStemmer()

new_text = "My two friends are getting married tomorrow and I could not be 
more excited for them"


words = word_tokenize(new_text)

for w in words:
    stems = [ps.stem(w)]
    print(stems)

我的输出：

['My']
['two']
['friend']
['are']
['get']
['marri']
['tomorrow']
['and']
['I']
['could']
['not']
['be']
['more']
['excit']
['for']
['them']

Answer 1

您可以改用列表推导。

print([ps.stem(w) for w in words])