python词云图背景_python词云图之WordCloud

1. 导入需要的包package

importmatplotlib.pyplot as pltfrom scipy.misc importimreadfrom wordcloud importWordCloud,STOPWORDSimport xlrd

2. 设置生成词云图的背景图片，最好是分辨率高且色彩边界分明的图片

defset_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")

plt.show()

4. 默认的停用词一般在：假如anaconda安装在D盘，则会在其目录：D：\Anaco