今天教你們用wrodcloud模塊來生成詞雲,我讀取了一篇小說並生成了詞雲,先看一下效果圖:python
效果圖一:bash
效果圖二: 微信
根據效果圖分析的仍是比較準確的,小說中的主人公就是「程理」,因此出現次數最多。圖中有兩種模式,一種是默認的模式,另外一種是本身添加圖片做爲背景。下面我就帶你們一塊兒來學習怎樣去生成詞雲!學習
對於新人來講安裝wordcloud模塊就是一大關,咱們通常都是經過Pycharm或者PIP安裝的,可是在安裝wordcloud時會提示錯誤,以下:字體
個人解決方法是直接訪問下面的網址,根據你的電腦和py版本下載對應的whl文件spa
https://www.lfd.uci.edu/~gohlke/pythonlibs/#wordcloud3d
下載完成後,把whl文件放在pip路徑下,輸入下面的代碼,記住代碼的whl文件名必定是適合你的文件名,我只是舉例:code
pip install wordcloud-1.4.1-cp36-cp36m-win32.whl
複製代碼
運行後會顯示安裝成功cdn
wordcloud的基本方法,看源碼以下:blog
各個參數你們應該一看就明白,另外wordcloud還有幾個其餘方法,好比fit_words,generate(text),to_file(filename)等等都是很是好理解的。
效果圖一的代碼:
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()
複製代碼
運行後:
但願對你們有所幫助!
你們能夠關注個人微信公衆號:「秦子帥」一個有質量、有態度的公衆號!