圖片發(fā)自簡書App
強大的word cloud庫湿右，還可以自定義背景圖片盏筐。
# -*- coding: utf-8 -*-
import matplotlib.pyplot as plt
import pickle
from wordcloud import WordCloud,STOPWORDS,ImageColorGenerator
import jieba
with open('HotelComments.txt') as fin:
for line in fin.readlines():
line = line.strip('\n')
text += ' '.join(jieba.cut(line))
text += ' '
fout = open('text.txt','wb')
pickle.dump(text,)
fout.close()
fr = open('text.txt','rb')
text = pickle.load(fr) # wordcloud必須傳入二進制文件
backgroud_Image = plt.imread('girl.jpg')
wc = WordCloud( background_color = 'white', # 設置背景顏色
mask = backgroud_Image, # 設置背景圖片
max_words = 2000, # 設置最大現實的字數
stopwords = STOPWORDS, # 設置停用詞
font_path = 'C:/Users/Windows/fonts/msyh.ttf',# 設置字體格式蟹漓，如不設置顯示不了中文
max_font_size = 50, # 設置字體最大值
random_state = 30, # 設置有多少種隨機生成狀態(tài)纳寂，即有多少種配色方案
)
wc.generate(text)
image_colors = ImageColorGenerator(backgroud_Image)
wc.recolor(color_func = image_colors)
plt.imshow(wc)
plt.axis('off')
plt.show()
