用Python制作词云

拥有回忆 提交于 2020-02-04 11:35:11

  用Python制作词云非常酷炫,先看张图。所有的词围绕在一张人物图像身上。

     要用python完成上图一样的词云,只要4步就可以完成。首先做好准备工作,需要准备一个文本文件,可以是歌词,小说的章节等。然后准备一张图片和需要用到的一个字体文件SimHei.ttf。当这些准备工作完成后,开始制作了。

第一步:读取文本文件

#1、读入文本数据
text = open(r'.\jay.txt','r').read()

第二步:用jieba库进行分词,需要导入jieba库

#2、分词
text_cut = jieba.cut(text)
result = ' '.join(text_cut) #把分开的词,重新连接起来

第三步:生成词云图,这里要用到WordCloud库和imread库

#3、生成词云图
#获取图片
imag=imread(r'.\jay1.jpg')
wc = WordCloud(
    #字体路径
    font_path=r'.\SimHei.ttf',
    #背景颜色
    background_color='white',
    #width=500,
    #height=350,
    max_font_size=50,
    #min_font_size=12,
    mask=imag
)
wc.generate(result)

第四步:重定向词云图文件或用 matplotlib库绘制出来

#4、重定向图片
wc.to_file(r'.\wordcloud.png') #创建词云图片

# 显示图片
plt.figure('jay')
plt.imshow(wc)
plt.axis('off') #关闭坐标轴
plt.show()

完成以上几个步骤,一个词云图就生成了。

以下是整个代码:

#matplotlib: 2D数学绘图库
import matplotlib.pyplot as plt
#jieba: 中文分词
import jieba
#WordCloud: 词云库
from wordcloud import WordCloud
#读取图片需要用到
from scipy.msic import imread
#from imageio import imread

#1、读入文本数据
text = open(r'.\jay.txt','r').read()
#2、分词
text_cut = jieba.cut(text)
result = ' '.join(text_cut) #把分开的词,重新连接起来
#3、生成词云图
#获取图片
imag=imread('jay1.jpg')
wc = WordCloud(
    #字体路径
    font_path=r'.\SimHei.ttf',
    #背景颜色
    background_color='white',
    #width=500,
    #height=350,
    max_font_size=50,
    #min_font_size=12,
    mask=imag
)
wc.generate(result)
#4、重定向图片
wc.to_file(r'.\wordcloud.png') #创建词云图片

# 显示图片
plt.figure('jay')
plt.imshow(wc)
plt.axis('off') #关闭坐标轴
plt.show()

 

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!