1. 程式人生 > >python(wordcloud)實現中文詞雲

python(wordcloud)實現中文詞雲

bold pytho 作圖 back 垂直 背景 數值 內置 顯示

#   這是一個處理圖像的函數
from scipy.misc import imread
from wordcloud import WordCloud,STOPWORDS,ImageColorGenerator
import matplotlib.pylab as plt

# 解析圖片
back_color = imread("./veer-141001498.png")
# 設置字體路徑
font = "C:\Windows\Fonts\STXINGKA.TTF"
wc = WordCloud(background_color="white", # 背景顏色
max_words=500, # 最大詞數
mask=back_color, # 掩膜,產生詞雲背景的區域,以該參數值作圖繪制詞雲,這個參數不為空時,width,height會被忽略
max_font_size=80, # 顯示字體的最大值
stopwords=STOPWORDS.add("差評"), # 使用內置的屏蔽詞,再添加一個
font_path=font, # 解決顯示口字型亂碼問題,可進入C:/Windows/Fonts/目錄更換字體
random_state=42, # 為每一詞返回一個PIL顏色
prefer_horizontal=10) # 調整詞雲中字體水平和垂直的多少
# 打開詞源的文本文件
text = open("./comments.txt","r",encoding="utf-8").read()

# 生成詞雲
wc.generate(text)
# 從背景圖片生成顏色值
image_colors = ImageColorGenerator(back_color)
# 顯示圖片
plt.imshow(wc)
# 關閉坐標軸
plt.axis("off")
# 繪制詞雲
plt.figure()

plt.imshow(wc.recolor(color_func=image_colors))

plt.axis("off")
# 保存圖片
wc.to_file("text2.png")



# 以下為效果圖
技術分享圖片

python(wordcloud)實現中文詞雲