Python如何生成詞雲(詳細分析)

秦子帥發表於2018-05-31

前言

今天教大家用wrodcloud模組來生成詞雲,我讀取了一篇小說並生成了詞雲,先看一下效果圖:

效果圖一:

image

效果圖二:

Python如何生成詞雲(詳細分析)

image

根據效果圖分析的還是比較準確的,小說中的主人公就是“程理”,所以出現次數最多。圖中有兩種模式,一種是預設的模式,另一種是自己新增圖片作為背景。下面我就帶大家一起來學習怎樣去生成詞雲!

wordcloud的安裝

對於新人來說安裝wordcloud模組就是一大關,我們一般都是通過Pycharm或者PIP安裝的,但是在安裝wordcloud時會提示錯誤,如下:

image

我的解決方法是直接訪問下面的網址,根據你的電腦和py版本下載對應的whl檔案

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

image

下載完成後,把whl檔案放在pip路徑下,輸入下面的程式碼,記住程式碼的whl檔名一定是適合你的檔名,我只是舉例:


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

複製程式碼

執行後會顯示安裝成功

Python如何生成詞雲(詳細分析)

wordcloud的簡單用法

wordcloud的基本方法,看原始碼如下:

Python如何生成詞雲(詳細分析)

各個引數大家應該一看就明白,另外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()

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

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

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

複製程式碼

執行後:

Python如何生成詞雲(詳細分析)

效果圖二,給詞雲加上圖片背景,程式碼如下:


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))
   # 結巴分詞,生成字串,如果不通過分詞,無法直接生成正確的中文詞雲,感興趣的朋友可以去查一下,有多種分詞模式
   #Python join() 方法用於將序列中的元素以指定的字元連線生成一個新的字串。
   cut_text = " ".join(jieba.cut(f))

   wordcloud = WordCloud(
       # 設定字型,不然會出現口字亂碼,文字的路徑是電腦的字型一般路徑,可以換成別的
       font_path="C:/Windows/Fonts/simfang.ttf",
       background_color="white",
       # mask引數=圖片背景,必須要寫上,另外有mask引數再設定寬高是無效的
       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()

複製程式碼

執行後:

image

希望對大家有所幫助!

大家可以關注我的微信公眾號:「秦子帥」一個有質量、有態度的公眾號!

Python如何生成詞雲(詳細分析)

相關文章