Python 词云可视化

安装jieba和wordcloud,cmd进入终端

pip install jieba
pip install wordcloud

Code:

import jieba
from matplotlib import pyplot as plt
from wordcloud import WordCloud
from PIL import Image
import numpy as np
#r''单引号里面不需要转义
path = r'E:Data'
font = r'C:WindowsFontssimkai.ttf'#电脑自带的字体
def tcg(texts):
    cut = jieba.cut(texts)  #分词
    string = ' '.join(cut)
    return string
text = (open(path+r'三国演义.txt','r',encoding='utf-8')).read()
string=tcg(text)

img = Image.open(path+r'background.jpg') #打开图片
img_array = np.array(img) #将图片装换为数组
stopword=['']  #设置停止词，也就是你不想显示的词
wc = WordCloud(
    background_color='white',
    width=1000,
    height=800,
    mask=img_array, #设置背景图片
    font_path=font,
    stopwords=stopword
)
wc.generate_from_text(string)#绘制图片
plt.imshow(wc)
plt.axis('off')
plt.show()  #显示图片
wc.to_file(path+r'beautifulcloud.png')  #保存图片

注：三国演义.txt文件路径需要根据自己文件所在位置进行修改

background.jpg

beautifulcloud.png

importError: cannot import name 'WordCloud' from 'wordcloud'

安装了wordcloud,遇到这个报错,可以这个程序的名字修改为其他的，不能以wordcloud.py命名