import jieba
import imageio
import wordcloud


# 讀取txt文本
with open('resume.txt','r',encoding='utf-8') as f:
  text = f.read()

# 簡單處理文本，刪除空格等多余字符
text = text.replace(' ','').replace('\t','').replace('\n','')
# 使用jieba分隔詞語
wordlist = jieba.lcut(text)  # ['aa','bb','cc'...]
# print(wordlist)
# 用空格連接各個詞語，又形成一個大字符串
string = ' '.join(wordlist)  # 'aa bb cc'

# 讀取圖片
image = imageio.imread('ding.jpg')


# 生成詞云圖片
# 先實例化一個詞云對象
wc = wordcloud.WordCloud(width=image.shape[0], # 詞云圖寬度同原圖片寬度
            height=image.shape[1],
            background_color='white', # 背景顏色白色
            font_path='msyh.ttc', # 指定字體路徑，微軟雅黑，可從win自帶的字體庫中找
            mask=image,  # mask 指定詞云形狀圖片，默認為矩形
            scale=3)  # 默認為1，越大越清晰
# 再給詞云輸入文字
wc.generate(string)
# 保存成文件
wc.to_file('wordcloud21.png')