Python如何生成詞云(詳解)

Juven 發(fā)布于2019-07-30 16:44 / 2336人閱讀

摘要：前言今天教大家用模塊來生成詞云，我讀取了一篇小說并生成了詞云，先看一下效果圖效果圖一效果圖二根據(jù)效果圖分析的還是比較準(zhǔn)確的，小說中的主人公就是程理，所以出現(xiàn)次數(shù)最多。

前言

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

效果圖一：

效果圖二：

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

wordcloud的安裝

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

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

https://www.lfd.uci.edu/~gohl...

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

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

運(yùn)行后會(huì)顯示安裝成功

wordcloud的簡(jiǎn)單用法

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

各個(gè)參數(shù)大家應(yīng)該一看就明白，另外wordcloud還有幾個(gè)其他方法，比如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è)置字體，不然會(huì)出現(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()

運(yùn)行后：

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

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() 方法用于將序列中的元素以指定的字符連接生成一個(gè)新的字符串。
? ?cut_text =?" ".join(jieba.cut(f))

? ?wordcloud = WordCloud(
? ? ? ?# 設(shè)置字體，不然會(huì)出現(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()

運(yùn)行后：

如果有不懂的朋友可以直接在公眾號(hào)上加我的微信，希望對(duì)大家有所幫助！
大家可以關(guān)注我的微信公眾號(hào)：「秦子帥」一個(gè)有質(zhì)量、有態(tài)度的公眾號(hào)！