#導入需要模塊
import jieba
import numpy as np 
import matplotlib.pyplot as plt 
from PIL import Image 
from wordcloud import WordCloud, STOPWORDS, ImageColorGenerator

text_road=str(input('請輸入文章的路徑：'))
picture_road=str(input('請輸入圖片的路徑：'))

#加載需要分析的文章
text = open(text_road,'r',encoding='utf-8').read()

#對文章進行分詞
wordlist_after_jieba = jieba.cut(text, cut_all=False)
wl_space_split = " ".join(wordlist_after_jieba)

#讀取照片通過numpy.array函數(shù)將照片等結(jié)構(gòu)數(shù)據(jù)轉(zhuǎn)化為np-array
mask=np.array(Image.open(picture_road))

#選擇屏蔽詞，不顯示在詞云里面
stopwords = set(STOPWORDS)
#可以加多個屏蔽詞
stopwords.add("<br/>")

#創(chuàng)建詞云對象
wc = WordCloud(
    background_color="white",
    font_path='/Library/Fonts/Arial Unicode.ttf',
    max_words=1000, # 最多顯示詞數(shù)
    mask=mask, 
    stopwords=stopwords,
    max_font_size=100 # 字體最大值
    )

#生成詞云
wc.generate(text)

#從背景圖建立顏色方案
image_colors =ImageColorGenerator(mask) 

#將詞云顏色設(shè)置為背景圖方案
wc.recolor(color_func=image_colors) 

#顯示詞云
plt.imshow(wc,interpolation='bilinear')

#關(guān)閉坐標軸
plt.axis("off")

#顯示圖像
plt.show()

#保存詞云
wc.to_file('詞云圖.png')