使用python的NLTK计算动词,名词和其他词性

时间:2012-05-20 15:41:15

标签: python nlp tagging nltk part-of-speech

我有多个文本,我想基于各种词性的使用来创建它们的配置文件,如名词和动词。基本上,我需要计算每个词性的使用次数。

我已对文字进行了标记,但我不确定如何更进一步:

tokens = nltk.word_tokenize(text.lower())
text = nltk.Text(tokens)
tags = nltk.pos_tag(text)

如何将每个词性的计数保存到变量中?

1 个答案:

答案 0 :(得分:30)

pos_tag方法会返回一个(标记,标记)对的列表:

tagged = [('the', 'DT'), ('dog', 'NN'), ('sees', 'VB'), ('the', 'DT'), ('cat', 'NN')] 

如果您使用的是Python 2.7或更高版本,那么您只需使用:

即可
>>> from collections import Counter
>>> counts = Counter(tag for word,tag in tagged)
>>> counts
Counter({'DT': 2, 'NN': 2, 'VB': 1})

规范化计数(给出每个的比例):

>>> total = sum(counts.values())
>>> dict((word, float(count)/total) for word,count in counts.items())
{'DT': 0.4, 'VB': 0.2, 'NN': 0.4}

请注意,在旧版本的Python中,您必须自己实现Counter

>>> from collections import defaultdict
>>> counts = defaultdict(int)
>>> for word, tag in tagged:
...  counts[tag] += 1

>>> counts
defaultdict(<type 'int'>, {'DT': 2, 'VB': 1, 'NN': 2})