scikit-learn：CountVectorizer提取词频

CountVectorizer详解

最新推荐文章于 2025-09-21 22:52:10 发布

转载最新推荐文章于 2025-09-21 22:52:10 发布 · 3.4k 阅读

Python 同时被 2 个专栏收录

67 篇文章

订阅专栏

Python；数据分析

15 篇文章

订阅专栏

本文深入解析了sklearn库中CountVectorizer的使用方法与参数配置，详细介绍了如何通过此工具进行文本预处理，包括去除停用词、转换大小写、提取n-gram等，以及如何根据需求调整参数以优化文本特征提取过程。

http://scikit-learn.org/stable/modules/generated/sklearn.feature_extraction.text.CountVectorizer.html#sklearn.feature_extraction.text.CountVectorizer

sklearn.feature_extraction.text.CountVectorizer(

input=u'content', encoding=u'utf-8', decode_error=u'strict',strip_accents=None, lowercase=True, preprocessor=None, tokenizer=None, stop_words=None,token_pattern=u'(?u)\b\w\w+\b', ngram_range=(1, 1), analyzer=u'word', max_df=1.0, min_df=1,max_features=None, vocabulary=None, binary=False, dtype=)

作用：Convert a collection of text documents to a matrix of token counts（计算词汇的数量，即tf）；结果由 scipy.sparse.coo_matrix进行稀疏表示。

看下参数就知道CountVectorizer在提取tf时都做了什么：

strip_accents : {‘ascii’, ‘unicode’, None}：是否除去“音调”，不知道什么是“音调”？看：http://textmechanic.com/?reqp=1&reqr=nzcdYz9hqaSbYaOvrt==

lowercase : boolean, True by default：计算tf前，先将所有字符转化为小写。这个参数一般为True。

preprocessor : callable or None (default)：复写the preprocessing (string transformation) stage，但保留tokenizing and n-grams generation steps.这个参数可以自己写。

tokenizer : callable or None (default)：复写the string tokenization step，但保留preprocessing and n-grams generation steps.这个参数可以自己写。

stop_words : string {‘english’}, list, or None (default)：如果是‘english’, a built-in stop word list for English is used。如果是a list，那么最终的tokens中将去掉list中的所有的stop word。如果是None, 不处理停顿词；但参数 max_df可以设置为 [0.7, 1.0) 之间，进而根据intra corpus document frequency(df) of terms自动detect and filter stop words。这个参数要根据自己的需求调整。

token_pattern : string：正则表达式，默认筛选长度大于等于2的字母和数字混合字符（select tokens of 2 or more alphanumeric characters ），参数analyzer设置为word时才有效。

ngram_range : tuple (min_n, max_n)：n-values值得上下界，默认是ngram_range=(1, 1)，该范围之内的n元feature都会被提取出来！这个参数要根据自己的需求调整。

analyzer : string, {‘word’, ‘char’, ‘char_wb’} or callable：特征基于wordn-grams还是character n-grams。如果是callable是自己复写的从the raw, unprocessed input提取特征的函数。

max_df : float in range [0.0, 1.0] or int, default=1.0： min_df : float in range [0.0, 1.0] or int, default=1：按比例，或绝对数量删除df超过max_df或者df小于min_df的word tokens。有效的前提是参数vocabulary设置成Node。

max_features : int or None, default=None：选择tf最大的max_features个特征。有效的前提是参数vocabulary设置成Node。

vocabulary : Mapping or iterable, optional：自定义的特征word tokens，如果不是None，则只计算vocabulary中的词的tf。还是设为None靠谱。

binary : boolean, default=False：如果是True，tf的值只有0和1，表示出现和不出现，useful for discrete probabilistic models that model binary events rather than integer counts.。

dtype : type, optional：Type of the matrix returned by fit_transform() or transform().。

结论： CountVectorizer提取tf都做了这些：去音调、转小写、去停顿词、在word（而不是character，也可自己选择参数）基础上提取所有ngram_range范围内的特征，同时删去满足“max_df, min_df,max_features”的特征的tf。当然，也可以选择tf为binary。

最后看下两个函数：

fit(raw_documents[, y]) Learn a vocabulary dictionary of all tokens in the raw documents.

fit_transform(raw_documents[, y]) Learn the vocabulary dictionary and return term-document matrix.

转自：https://blog.youkuaiyun.com/mmc2015/article/details/46866537?utm_source=copy

from sklearn.feature_extraction.text import CountVectorizer

texts=["dog cat fish","dog cat cat","fish bird", 'bird']
cv = CountVectorizer()
cv_fit=cv.fit_transform(texts)

print((cv.get_feature_names()))
print(cv.vocabulary_)  # 索引

print((cv_fit.toarray()))
print('对应特征总个数：', (cv_fit.toarray().sum(axis=0)))  # .sum(axis=0)按列求和