这篇文章主要介绍了python基于WordCloud制作词云图,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
1. 导入需要的包package
import matplotlib.pyplot as plt
from scipy.misc import imread
from wordcloud import WordCloud,STOPWORDS
import xlrd
2. 设置生成词云图的背景图片,最好是分辨率高且色彩边界分明的图片
def set_background(picpath):
back_coloring = imread(picpath)# 设置背景图片,png等图片格式
return back_coloring
3. 创建词云图:WordCloud
def create_word_cloud(txt_str, back_coloring): #txt_str表示导入的是字符串格式数据,#back_color表示的是背景图片位置
print('---- 根据词频,开始生成词云! ----')
font = r'C:\Windows\Fonts\simsun.ttc' #加载显示字体
wc = WordCloud(
font_path=font,
collocations=False, # 去重,如果不加,词云图会显示相同的词
stopwords=STOPWORDS, #加载停用词,如果不自己指定,则会加载默认的停用词
max_words=100,
width=2000,
height=1200,
# background_color='white',
mask=back_coloring,
)
wordcloud = wc.generate(txt_str)
# 写词云图片
wordcloud.to_file(".\wordcloud_test.png")
# 显示词云文件
plt.imshow(wordcloud)
plt.axis("off