暑假实习|机器学习第四天

使用python中的jieba库识别文本，生成词云图

识别文本：

import jieba
from wordcloud import WordCloud, ImageColorGenerator
import matplotlib.pyplot as plt
from PIL import Image
import numpy as np

text = "如果没有以后如果平行时空那些我不同人生的我会以什么方式哭过"
#  `jieba.cut()`函数用于将文本进行分词，返回一个可迭代的生成器
#  可以使用`join()`函数将分词结果连接成字符串，并使用空格进行分隔
seg_list = jb.cut(text, cut_all=True)  # 全模式
print("/ ".join(seg_list))
seg_list = jb.cut(text, cut_all=False)  # 精确模式（默认）
print("/ ".join(seg_list))
seg_list = jb.cut_for_search(text)  # 搜索引擎模式
print("/ ".join(seg_list))

 生成词云图（规则图形）

# 读取文本文件  规则形状词云图
with open('D:\桌面文件\song.txt', 'r', encoding='utf-8') as f:
    text = f.read()  # 使用jieba进行中文分词
seg_list = jieba.cut(text)
seg_text = " ".join(seg_list)  # 生成词云图
wordcloud = WordCloud(font_path='simhei.ttf', background_color='white').generate(seg_text)  # 显示词云图  规则图像
plt.imshow(wordcloud, interpolation='bilinear')
plt.axis('off')
plt.show()

运行结果：

 

 

生成不规则图形词云图

读取文本文件  不规则图像词云图
with open('D:\桌面文件\song.txt', 'r', encoding='utf-8') as f:
    text = f.read()  # 使用jieba进行中文分词
seg_list = jieba.cut(text)
seg_text = " ".join(seg_list)
mask = np.array(Image.open('D:\桌面文件\cloud.png'))  # 读取自定义形状图像
wordcloud = WordCloud(font_path='simhei.ttf', background_color='white', mask=mask).generate(seg_text)  # 生成词云图
image_colors = ImageColorGenerator(mask)  # 根据形状图像生成颜色
plt.imshow(wordcloud.recolor(color_func=image_colors), interpolation='bilinear')
plt.axis('off')
plt.show()

这里需要先导入不规则图形的形状，例如：

 

 生成结果为：

 这里可以看到，导入的模板是黑色的，所以生成的也是黑色的

导入彩色模板试试

运行结果

 

生成的结果就是彩色的了