欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

python实现词云图

程序员文章站 2022-05-27 15:15:49
...

先看一下成果图:
python实现词云图

图片给大家,从某一篇帖子找的。

python实现词云图


需要使用的的库:
matplotlib.pyplot       绘图展示
jeiba            分词
wordcloud.WordCloud     绘制词云
numpy           制作背景图
PIL.Image           制作背景图

不知道说什么,说明全在代码里:

from matplotlib import pyplot as plt
import jieba
from wordcloud import WordCloud
import numpy as np      # 用于制作背景图
from PIL import Image       # 用于制作背景图


# 导入文字
text = open('gushi.txt', 'r', encoding='utf-8').read()
# 分词
cut_text = jieba.cut(text)
# 将分词连接成一个字符串
result = ' '.join(cut_text)
# 制作背景图
bg_img = np.array(Image.open('./feng.png'))
# 词云图初始化
wc = WordCloud(
    font_path='C:\WINDOWS\Fonts\SIMLI.TTF',
    background_color='white',
    width=500,
    height=400,
    max_font_size=50,
    min_font_size=10,
    # colormap='Greens',          # 文字限定配色, 有兴趣可以试一下
    mask=bg_img,        # 背景图片
    contour_width=1,		# 背景边框宽度
    contour_color='yellow',		# 背景边框颜色
    # mode='RGBA',		# 制作成png透明背景图片,不能与contour共存
)
# 制作词云图
wc.generate(result)
wc.to_file('./gushi.png')		# 保存到图片
# 显示图片
plt.figure('gushi')		# 标题
plt.imshow(wc)
plt.axis('off')       # 关闭坐标轴
plt.show()
相关标签: python