用Python製作詞雲

  用Python製作詞雲非常酷炫,先看張圖。所有的詞圍繞在一張人物圖像身上。

     要用python完成上圖一樣的詞雲,只要4步就可以完成。首先做好準備工作,需要準備一個文本文件,可以是歌詞,小說的章節等。然後準備一張圖片和需要用到的一個字體文件SimHei.ttf。當這些準備工作完成後,開始製作了。

第一步:讀取文本文件

#1、讀入文本數據
text = open(r'.\jay.txt','r').read()

第二步:用jieba庫進行分詞,需要導入jieba庫

#2、分詞
text_cut = jieba.cut(text)
result = ' '.join(text_cut) #把分開的詞,重新連接起來

第三步:生成詞雲圖,這裏要用到WordCloud庫和imread庫

#3、生成詞雲圖
#獲取圖片
imag=imread(r'.\jay1.jpg')
wc = WordCloud(
    #字體路徑
    font_path=r'.\SimHei.ttf',
    #背景顏色
    background_color='white',
    #width=500,
    #height=350,
    max_font_size=50,
    #min_font_size=12,
    mask=imag
)
wc.generate(result)

第四步:重定向詞雲圖文件或用 matplotlib庫繪製出來

#4、重定向圖片
wc.to_file(r'.\wordcloud.png') #創建詞雲圖片

# 顯示圖片
plt.figure('jay')
plt.imshow(wc)
plt.axis('off') #關閉座標軸
plt.show()

完成以上幾個步驟,一個詞雲圖就生成了。

以下是整個代碼:

#matplotlib: 2D數學繪圖庫
import matplotlib.pyplot as plt
#jieba: 中文分詞
import jieba
#WordCloud: 詞雲庫
from wordcloud import WordCloud
#讀取圖片需要用到
from scipy.msic import imread
#from imageio import imread

#1、讀入文本數據
text = open(r'.\jay.txt','r').read()
#2、分詞
text_cut = jieba.cut(text)
result = ' '.join(text_cut) #把分開的詞,重新連接起來
#3、生成詞雲圖
#獲取圖片
imag=imread('jay1.jpg')
wc = WordCloud(
    #字體路徑
    font_path=r'.\SimHei.ttf',
    #背景顏色
    background_color='white',
    #width=500,
    #height=350,
    max_font_size=50,
    #min_font_size=12,
    mask=imag
)
wc.generate(result)
#4、重定向圖片
wc.to_file(r'.\wordcloud.png') #創建詞雲圖片

# 顯示圖片
plt.figure('jay')
plt.imshow(wc)
plt.axis('off') #關閉座標軸
plt.show()

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章