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

前言

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

效果圖一:bash

image

效果圖二: 微信

image

根據效果圖分析的仍是比較準確的,小說中的主人公就是「程理」,因此出現次數最多。圖中有兩種模式,一種是默認的模式,另外一種是本身添加圖片做爲背景。下面我就帶你們一塊兒來學習怎樣去生成詞雲!學習

wordcloud的安裝

對於新人來講安裝wordcloud模塊就是一大關,咱們通常都是經過Pycharm或者PIP安裝的,可是在安裝wordcloud時會提示錯誤,以下:字體

image

個人解決方法是直接訪問下面的網址,根據你的電腦和py版本下載對應的whl文件spa

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

image

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

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

複製代碼

運行後會顯示安裝成功cdn

wordcloud的簡單用法

wordcloud的基本方法,看源碼以下:blog

各個參數你們應該一看就明白,另外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()

複製代碼

運行後:

image

但願對你們有所幫助!

你們能夠關注個人微信公衆號:「秦子帥」一個有質量、有態度的公衆號!

相關文章
相關標籤/搜索