快捷導(dǎo)航

python利用WordCloud模塊實(shí)現(xiàn)詞云繪制

更新時(shí)間：2023年10月23日 09:32:50 作者：微小冷

wordcloud是詞云繪圖模塊,封裝了WordCloud詞云類,是詞云的基本載體,下面小編就來(lái)和大家詳細(xì)講講如何利用WordCloud模塊實(shí)現(xiàn)詞云繪制吧,希望對(duì)大家有所幫助

初步認(rèn)識(shí)

wordcloud是詞云繪圖模塊，封裝了WordCloud詞云類，是詞云的基本載體。在新建一個(gè)詞云之后，通過(guò)generate裝載用以生成詞云的字符串，最后用to_file把詞云圖保存到文件中，例如

from wordcloud import WordCloud

w = WordCloud()
txt = "tinycool cool tiny Python WordCloud word cloud word not world"
w.generate(txt)
w.to_file("pywordcloud.png")

效果如下

基本參數(shù)

僅就這張圖像來(lái)說(shuō)，我們能感受到的特性大概有

詞云圖的寬度、高度
背景顏色，文字顏色
字體
單詞長(zhǎng)度，單詞個(gè)數(shù)

這些特性可受到下列參數(shù)調(diào)控

參數(shù)	說(shuō)明	備注
font_path	字體文件路徑	otf或ttf文件
width	詞云寬度	默認(rèn)400
height	詞云高度	默認(rèn)200
min_font_size	最小文字尺寸	默認(rèn)4
max_font_size	最大文字尺寸	默認(rèn)為圖像高度
relative_scaling	詞頻對(duì)尺寸的影響	默認(rèn)"auto"
font_step	字體步長(zhǎng)	默認(rèn)1
max_words	最大單詞數(shù)	默認(rèn)200
min_word_length	最短單詞長(zhǎng)度	默認(rèn)0
background_color	背景色	默認(rèn)"black"
mode	顏色格式	默認(rèn)RGB
colormap	顏色映射
repeat	是否重復(fù)單詞	默認(rèn)False

下面對(duì)這些參數(shù)稍作更改，并

# txt即本文前面所有文字，因?yàn)樘嗑筒粚懥?
txt = r'''
'''
dct = dict(width=600, height=300,
    font_path = r"C:\Windows\Fonts\simhei.ttf",
    min_font_size = 10,
    max_font_size = 100,
    max_words = 20,
    min_word_length = 2,
    background_color = "white",
    colormap = "jet")

w = WordCloud(**dct)
w.generate(txt)
w.to_file("pywordcloud.png")

效果如下，由于最多只顯示20個(gè)單詞，而且尺寸最小是10，最大100，所以不太好看。

掩模參數(shù)

所謂掩模，就是詞云組成的圖案，主要由以下參數(shù)調(diào)控。

參數(shù)	說(shuō)明	備注
mask	掩模數(shù)組	默認(rèn)None
contour_width	掩模寬度	默認(rèn)為0
contour_color	掩模顏色	默認(rèn)"black"

掩模數(shù)組由0和1構(gòu)成，詞云只在掩模數(shù)組為1的位置顯示。下面畫一個(gè)橢圓作為掩模數(shù)組，橢圓圓周的方程為

import numpy as np
y, x = np.indices([300,600])
delta = np.sqrt(280**2-(2*y-300)**2)
mask = (x<300+delta) & (x>300-delta)
mask = (1-mask)*254+1

w = WordCloud( mask = mask, contour_width=100,
    font_path = r"C:\Windows\Fonts\simhei.ttf")
w.generate(txt)
w.to_file("pywordcloud.png")

效果如下