快捷導(dǎo)航

python根據(jù)文本生成詞云圖代碼實(shí)例

更新時(shí)間：2019年11月15日 10:57:42 作者：夢(mèng)小冷

這篇文章主要介紹了python根據(jù)文本生成詞云圖代碼實(shí)例,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下

效果

代碼

from wordcloud import WordCloud
import codecs
import jieba
#import jieba.analyse as analyse
from scipy.misc import imread
import os
from os import path
import matplotlib.pyplot as plt
from PIL import Image, ImageDraw, ImageFont


# 繪制詞云
def draw_wordcloud():
  #讀入一個(gè)txt文件
  comment_text = open('data.txt','r',encoding='utf-8').read()
  #結(jié)巴分詞，生成字符串，如果不通過分詞，無法直接生成正確的中文詞云
  cut_text = " ".join(jieba.cut(comment_text))
  d = path.dirname(__file__) # 當(dāng)前文件文件夾所在目錄
  color_mask = imread("bilibili.jpg") # 讀取背景圖片
  cloud = WordCloud(
    #設(shè)置字體，不指定就會(huì)出現(xiàn)亂碼
    font_path="C:/Windows/Fonts/STXINGKA.TTF",
    #font_path=path.join(d,'simsun.ttc'),
    #設(shè)置背景色
    background_color='white',
    #詞云形狀
    mask=color_mask,
    #允許最大詞匯
    max_words=2000,
    #最大號(hào)字體
    max_font_size=100
  )
  word_cloud = cloud.generate(cut_text) # 產(chǎn)生詞云
  word_cloud.to_file("bilibilirankword.jpg") #保存圖片
  # 顯示詞云圖片
  plt.imshow(word_cloud)
  plt.axis('off')
  plt.show()

if __name__ == '__main__':
  draw_wordcloud()

步驟以及注意點(diǎn)

1.首先要導(dǎo)入對(duì)應(yīng)的庫

2.文字會(huì)根據(jù)背景圖而分布

3.背景圖的選擇對(duì)于最終效果很重要，選一個(gè)盡量大一點(diǎn)的背景圖，這個(gè)背景圖隨便找就好，任何圖片都可以不過最好大一些，圖案不太復(fù)雜

4.如果字號(hào)小的話，可以調(diào)整字體，或者換個(gè)背景圖

5.font_path要使用自己的字體庫，不過你的電腦是windows的話，應(yīng)該不用改，用我這個(gè)就可以，除此之外呢，這個(gè)字體如果是別的的話，可能會(huì)出現(xiàn)

中文不能產(chǎn)生詞云圖的現(xiàn)象，總之用我的就好了。

以上就是本文的全部內(nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章: