前言
今天教大家用wrodcloud模組來生成詞雲,我讀取了一篇小說並生成了詞雲,先看一下效果圖:
效果圖一:
效果圖二:
根據效果圖分析的還是比較準確的,小說中的主人公就是“程理”,所以出現次數最多。圖中有兩種模式,一種是預設的模式,另一種是自己新增圖片作為背景。下面我就帶大家一起來學習怎樣去生成詞雲!
wordcloud的安裝
對於新人來說安裝wordcloud模組就是一大關,我們一般都是通過Pycharm或者PIP安裝的,但是在安裝wordcloud時會提示錯誤,如下:
我的解決方法是直接訪問下面的網址,根據你的電腦和py版本下載對應的whl檔案
https://www.lfd.uci.edu/~gohlke/pythonlibs/#wordcloud
下載完成後,把whl檔案放在pip路徑下,輸入下面的程式碼,記住程式碼的whl檔名一定是適合你的檔名,我只是舉例:
pip install wordcloud-1.4.1-cp36-cp36m-win32.whl
複製程式碼
執行後會顯示安裝成功
wordcloud的簡單用法
wordcloud的基本方法,看原始碼如下:
各個引數大家應該一看就明白,另外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()
複製程式碼
執行後:
效果圖二,給詞雲加上圖片背景,程式碼如下:
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()
複製程式碼
執行後:
希望對大家有所幫助!
大家可以關注我的微信公眾號:「秦子帥」一個有質量、有態度的公眾號!