python学习笔记之制作词云图

最新推荐文章于 2023-12-22 12:01:39 发布

碎小流

最新推荐文章于 2023-12-22 12:01:39 发布

阅读量297

点赞数

文章标签： python

本文链接：https://blog.csdn.net/byhjnwei/article/details/105689995

版权

python学习笔记之制作词云图

今天学习使用Python来制作简单的词云图。

先放上全部代码：

from matplotlib import pyplot as plt 
from wordcloud import WordCloud
import jieba
from PIL import Image
import numpy as np
from imageio import  imread

def gettext():
    f=open('三国演义.txt','r',encoding='gbk', errors='ignore').read()
    return f
sanguo=gettext()
words=jieba.lcut(sanguo)  #分词
string=' '.join(words)   #使用空格连接分词

#设置背景图片
# img=Image.open('map.jpg')
# img_array=np.array(img)
#or
img_array=imread('map.jpg')

#设置停止词
stopword=[]

#设置词云图大小
font=r'C:\Windows\Fonts\simkai.ttf' #设置字体路径

wc=WordCloud(font_path=font,
background_color='white',
width=1200,
height=800,
mask=img_array,   #设置背景图片
stopwords=stopword
)
wc.generate_from_text(string)   #绘制词云图
#wc.generate(string)   #效果类似

wc.to_file('new.png')  #保存图片

plt.imshow(wc)   #显示图片
plt.axis('off')
#plt.figure()
plt.show()