python学习笔记之制作词云图

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()