擅长:python、mysql、java
<pre><code>|col|
['Sushi Bars', 'Restaurants']
['Burgers', 'Fast Food', 'Restaurants']
wnl = WordNetLemmatizer()
</code></pre>
<p>下面创建一个函数,该函数接受单词列表并返回元素化单词列表。这应该管用。</p>
<pre><code>def lemmatize(s):
'''For lemmatizing the word
'''
s = [wnl.lemmatize(word) for word in s]
return s
dataset = dataset.assign(col_lemma = dataset.col.apply(lambda x: lemmatize(x))
</code></pre>