利用sklearn进行字典&文本的特征提取

写在前面

这篇博客主要内容：

1. 应用DictVectorizer实现对类别特征进行数值化、离散化
2. 应用CountVectorizer实现对文本特征进行数值化

特征提取API

sklearn.feature_extraction

字典特征提取

做用：对字典数据进行特征值化

• sklearn.feature_extraction.DictVectorizer(sparse=True,…)
  • DictVectorizer.fit_transform(X) X:字典或者包含字典的迭代器返回值：返回sparse矩阵
  • DictVectorizer.inverse_transform(X) X:array数组或者sparse矩阵 返回值:转换以前数据格式
  • DictVectorizer.get_feature_names() 返回类别名称

# 数据
[{'city': '北京','temperature':100}
{'city': '上海','temperature':60}
{'city': '深圳','temperature':30}]

# 代码
from sklearn.feature_extraction import DictVectorizer

def dict_demo():
    data = [{'city': '北京','temperature':100}, {'city': '上海','temperature':60}, {'city': '深圳','temperature':30}]
	# 一、实例化一个转换器类
    transfer  = DictVectorizer(sparse=False)
    # 二、调用fit_transform
    data_new = transfer.fit_transform(data)
    print("data_new：\n",data_new)
    # 打印特征名字
    print("特征名字：\n",transfer.get_feature_names())
    
    return None

注意DictVectorizer默认是true，输出为稀疏矩阵，false输出为普通矩阵

文本特征提取

做用：对文本数据进行特征值化

• sklearn.feature_extraction.text.CountVectorizer(stop_words=[])

  • 返回词频矩阵

• CountVectorizer.fit_transform(X) X:文本或者包含文本字符串的可迭代对象 返回值：返回sparse矩阵

• CountVectorizer.inverse_transform(X) X:array数组或者sparse矩阵 返回值:转换以前数据格

• CountVectorizer.get_feature_names() 返回值:单词列表

• sklearn.feature_extraction.text.TfidfVectorizer

# 数据
["life is short,i like python",
"life is too long,i dislike python"]

# 代码
from sklearn.feature_extraction.text import CountVectorizer

def count_demo():
    data = ["life is short,i like like python", "life is too long,i dislike python"]
    transfer  = CountVectorizer()
    data_new = transfer.fit_transform(data)
    print("data_new：\n",data_new.toarray())
    print("特征名字：\n",transfer.get_feature_names())
    return None

注意代码中的使用了toarray()，能够不加这个方法，再运行一下看看📑