Python练习（二）wordcloud生成词云

最新推荐文章于 2023-01-06 15:05:02 发布

eddieHoo

最新推荐文章于 2023-01-06 15:05:02 发布

阅读量2.1k

点赞数 1

分类专栏： Python

本文链接：https://blog.csdn.net/u011323949/article/details/102970777

版权

Python 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

1.前期准备

一个txt文档，一张背景图，一个ttf文件；

关键安装模块

pip install wordcloud
pip install jieba  //字符过滤用

2.编码

import numpy as np
from PIL import Image
import re
import jieba
from wordcloud import WordCloud,ImageColorGenerator,STOPWORDS
import  matplotlib.pyplot as plt

# 打开存放项目名称的txt文件
with open('content.txt','r',encoding='utf-8') as f:
    word= (f.read())
    f.close()

# 图片模板和字体
image=np.array(Image.open('bg.jpg')) 
font=r'simhei.ttf' 

# 去掉英文，保留中文 
resultword=re.sub("[A-Za-z0-9\[\`\~\!\@\#\$\^\&\*\(\)\=\|\{\}\'\:\;\'\,\[\]\.\<\>\/\?\~\。\@\#\\\&\*\%]", "",word) 
wordlist_after_jieba = jieba.cut(resultword) 
wl_space_split = " ".join(wordlist_after_jieba) 

# 设置停用词 
sw = set(STOPWORDS) 
sw.add("研发")
sw.add("系列")
sw.add("这里不多写了，根据自己情况添加")

# 关键一步
my_wordcloud = WordCloud(scale=4,font_path=font,mask=image,stopwords=sw,background_color='white',
                         max_words = 100,max_font_size = 60,random_state=20).generate(wl_space_split) 

#显示生成的词云 
plt.imshow(my_wordcloud)
plt.axis("off") 
plt.show() 

#保存生成的图片
my_wordcloud.to_file('result.jpg')