快捷導(dǎo)航

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

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

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

初步認識

wordcloud是詞云繪圖模塊，封裝了WordCloud詞云類，是詞云的基本載體。在新建一個詞云之后，通過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ù)

僅就這張圖像來說，我們能感受到的特性大概有

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

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

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

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

# txt即本文前面所有文字，因為太多就不寫了
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個單詞，而且尺寸最小是10，最大100，所以不太好看。

掩模參數(shù)

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

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

掩模數(shù)組由0和1構(gòu)成，詞云只在掩模數(shù)組為1的位置顯示。下面畫一個橢圓作為掩模數(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")

效果如下