怎么用Pyecharts制作詞云圖

前言

之前我們使用wordcloud庫制作了詞云圖，今天我們就來學(xué)習(xí)另外一種制作詞云圖的方法，那就是pyecharts庫，與wordcloud庫不同的是，pyecharts庫除了可以制作詞云圖外，還可以制作30+ 種常見圖表。

我們可以通過pip來安裝此庫。

pip install pyecharts

需要注意的是，pyecharts庫現(xiàn)在是有兩個版本的，分為v0.5.X 和 v1 兩個大版本，v0.5.X 和 v1 間不兼容，v1 是一個全新的版本，默認(rèn)安裝是v1版本，而且v，0.5.X已經(jīng)不再更新了，所以本次教程默認(rèn)使用v1版本的。其官方文檔鏈接為（https://pyecharts.org/#/zh-cn/intro）。

小試牛刀

我們參考官方文檔，就可以簡單寫出一個例子。

from pyecharts.charts import WordCloud

data = [('張三',67),('李四',43),('王五',20)]

wc = WordCloud()
wc.add(series_name="小試牛刀", data_pair=data)
wc.render_notebook()

 

實戰(zhàn)

這里需要注意的就是傳入的數(shù)據(jù)格式為[(word1, count1), (word2, count2)]。所以實際案例中，我們就要構(gòu)造成該格式的數(shù)據(jù)即可，還是以贅婿小說為例。

from pyecharts.charts import WordCloud
import jieba
import collections

fp = open('贅婿.txt','r')
text = fp.read().strip().replace('\n','').replace(' ','')
cut_words = list(jieba.cut(text))

stopwords = [line.strip() for line in open('hit_stopwords.txt','r').readlines()]
result = []
for cut_word in cut_words:
    if cut_word not in stopwords:
        result.append(cut_word)
        
data = collections.Counter(result)
data = data.most_common(200)


wc = WordCloud()
wc.add(series_name="贅婿", data_pair=data)
wc.render_notebook()