詳解Python如何生成詞云的方法

更新時間：2018年06月01日 14:10:51 作者：秦子帥

這篇文章主要介紹了詳解Python如何生成詞云的方法，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧

前言

今天教大家用wrodcloud模塊來生成詞云，我讀取了一篇小說并生成了詞云，先看一下效果圖：

效果圖一：

效果圖二：

根據(jù)效果圖分析的還是比較準確的，小說中的主人公就是“程理”，所以出現(xiàn)次數(shù)最多。圖中有兩種模式，一種是默認的模式，另一種是自己添加圖片作為背景。下面我就帶大家一起來學(xué)習(xí)怎樣去生成詞云！

wordcloud的安裝

對于新人來說安裝wordcloud模塊就是一大關(guān)，我們一般都是通過Pycharm或者PIP安裝的，但是在安裝wordcloud時會提示錯誤，如下：

我的解決方法是直接訪問下面的網(wǎng)址，根據(jù)你的電腦和py版本下載對應(yīng)的whl文件

https://www.lfd.uci.edu/~gohlke/pythonlibs/#wordcloud

下載完成后，把whl文件放在pip路徑下,輸入下面的代碼，記住代碼的whl文件名一定是適合你的文件名，我只是舉例：

pip install wordcloud-1.4.1-cp36-cp36m-win32.whl

運行后會顯示安裝成功

wordcloud的簡單用法

wordcloud的基本方法，看源碼如下：

各個參數(shù)大家應(yīng)該一看就明白，另外wordcloud還有幾個其他方法，比如fit_words，generate(text)，to_file(filename)等等都是非常好理解的。

wordcloud生成詞云

效果圖一的代碼：

from wordcloud import WordCloud
import matplotlib.pyplot as plt #繪制圖像的模塊
import jieba     #jieba分詞

path_txt='C://Users/Administrator/Desktop/all.txt'
f = open(path_txt,'r',encoding='UTF-8').read()

# 結(jié)巴分詞，生成字符串，wordcloud無法直接生成正確的中文詞云
cut_text = " ".join(jieba.cut(f))

wordcloud = WordCloud(
 #設(shè)置字體，不然會出現(xiàn)口字亂碼，文字的路徑是電腦的字體一般路徑，可以換成別的
 font_path="C:/Windows/Fonts/simfang.ttf",
 #設(shè)置了背景，寬高
 background_color="white",width=1000,height=880).generate(cut_text)

plt.imshow(wordcloud, interpolation="bilinear")
plt.axis("off")
plt.show()

運行后：

效果圖二，給詞云加上圖片背景，代碼如下：

from PIL import Image
from wordcloud import WordCloud, ImageColorGenerator
import matplotlib.pyplot as plt
import numpy as np
import jieba
def GetWordCloud():
 path_txt = 'C://Users/Administrator/Desktop/all.txt'
 path_img = "C://Users/Administrator/Desktop/timg.jpg"
 f = open(path_txt, 'r', encoding='UTF-8').read()
 background_image = np.array(Image.open(path_img))
 # 結(jié)巴分詞，生成字符串，如果不通過分詞，無法直接生成正確的中文詞云,感興趣的朋友可以去查一下，有多種分詞模式
 #Python join() 方法用于將序列中的元素以指定的字符連接生成一個新的字符串。
 cut_text = " ".join(jieba.cut(f))

 wordcloud = WordCloud(
  # 設(shè)置字體，不然會出現(xiàn)口字亂碼，文字的路徑是電腦的字體一般路徑，可以換成別的
  font_path="C:/Windows/Fonts/simfang.ttf",
  background_color="white",
  # mask參數(shù)=圖片背景，必須要寫上，另外有mask參數(shù)再設(shè)定寬高是無效的
  mask=background_image).generate(cut_text)
 # 生成顏色值
 image_colors = ImageColorGenerator(background_image)
 # 下面代碼表示顯示圖片
 plt.imshow(wordcloud.recolor(color_func=image_colors), interpolation="bilinear")
 plt.axis("off")
 plt.show()

if __name__ == '__main__':
 GetWordCloud()

運行后：