word_cloud是python的一個(gè)三方庫(kù),稱為詞云也叫做文字云,是根據(jù)文本中的詞頻,對(duì)內(nèi)容進(jìn)行可視化的匯總,可以用來(lái)繪制用戶畫(huà)像

下載與安裝

安裝word_cloud可以使用 python 自帶的 pip 工具來(lái)進(jìn)行

首先確保已經(jīng)安裝python與pip并且pip以及更新到最新版本.
這里使用的python版本為3.7,pip版本為19.2.1

word_cloud依賴于numpy與pillow,當(dāng)然如果只想將圖像保存成圖片文件,
使用matplotlib庫(kù)代替也可以,此處索性就將三個(gè)庫(kù)都安裝好.

pip install numpy
pip install pillow
pip install matplotlib

安裝完成后,再安裝word_cloud

pip install wordcloud

如果安裝失敗,可以去github主頁(yè)下載zip包

根據(jù)文本繪制

word_cloud的使用非常簡(jiǎn)單,一下的demo展示了如何從字符串中生成詞頻圖并保存的

# -*- coding: utf-8 -*-
from wordcloud import WordCloud

text = "dog cat fish bird cat cat dog"
wc = WordCloud()
wc.generate(text)
wc.to_file("1.png")

word_cloud從給定的text中按空格讀取單詞,出現(xiàn)次數(shù)越多的單詞,在生成的圖像越大,效果圖如下

默認(rèn)效果.png

控制繪制參數(shù)

上面的例子比較粗糙,為了更精細(xì)化的控制,word_cloud提供了大量參數(shù)用來(lái)控制圖像的生

利用屬性設(shè)置一下圖像,效果如下

from wordcloud import WordCloud

text = "dog cat fish bird cat cat dog"
wc = WordCloud(
    background_color='white',
    width=600,
)
wc.generate(text)
wc.to_file("1.png")

簡(jiǎn)單設(shè)置一下背景和長(zhǎng)度.png

中文亂碼問(wèn)題解決

在使用word_cloud處理中文時(shí),會(huì)出現(xiàn)亂碼,如下

from wordcloud import WordCloud

text = "張三 李四 王五"
wc = WordCloud(
    background_color='white',
    width=600,
)
wc.generate(text)
wc.to_file("1.png")

默認(rèn)字體中文亂碼.png

這是由于word_cloud自帶的字體DroidSansMono.ttf不支持中文
解決方法為使用支持中午跟的字體替代(注意版權(quán)),下面例子使用旁門(mén)正道字體

from wordcloud import WordCloud

text = "張三 李四 王五"
wc = WordCloud(
    background_color='white',
    font_path="PangMenZhengDaoBiaoTiTi-1.ttf"
)
wc.generate(text)
wc.to_file("1.png")

指定字體后.png

在指定背景圖上生成

使用numpy與matplotlib庫(kù),可以在指定背景圖上生成詞云

# -*- coding: utf-8 -*-
import numpy as np
import matplotlib.pyplot as plt
from PIL import Image
from wordcloud import WordCloud, ImageColorGenerator

text = "dog cat fish bird lino tiger monkey rabbit koala panda snake"
# 加載背景圖
color_mask = np.array(Image.open("f.jpeg"))
wc = WordCloud(
    mask=color_mask,background_color='white'
)
wc.generate(text)
image_colors = ImageColorGenerator(color_mask)
# 在只設(shè)置mask的情況下 會(huì)得到一個(gè)擁有圖片形狀的詞云 axis默認(rèn)為on 會(huì)開(kāi)啟邊框
plt.imshow(wc, interpolation="bilinear")
plt.axis("off")
plt.savefig("a.jpg")
# 直接在構(gòu)造函數(shù)中直接給顏色 這種方式詞云將會(huì)按照給定的圖片顏色布局生成字體顏色策略
plt.imshow(wc.recolor(color_func=image_colors), interpolation="bilinear")
plt.axis("off")
plt.savefig("b.jpg")

原始圖

原始圖

按輪廓生成.jpg

按輪廓按顏色生成.jpg

給定的詞數(shù)比較少,導(dǎo)致效果不是太明顯,增加一下text內(nèi)容即可

numpy與matplotlib的用法此處不做具體講解,僅最簡(jiǎn)單使用