天天看點

WordCloud中屏蔽詞+背景色設定

利用python的wordcloud包生成詞雲是一個很實用的技能,最近在繪制詞雲的過程中又學會了如何屏蔽不想要出現的詞及設定更豐富的背景圖,特别是背景顔色的設定很多資料都是一筆帶過,語焉不詳,好在經過痛苦的摸索之後終于被我找到了方法,原諒我激動的心情,啰嗦了這麼多(捂臉。。。)

from PIL import Image

import numpy as np

from wordcloud import WordCloud,STOPWORDS, ImageColorGenerator

from matplotlib import pyplot as plt

import jieba

with open('bigdata.txt', 'r') as f:

    text = f.read()

bigdata = " ".join(jieba.cut(text))  

cloud_mask = np.array(Image.open("data.png"))

#生成wordcloud對象

wc = WordCloud(background_color=(135,206,250),     #背景色,除了寫'blue'等還可以調整參數來設定更為豐富的色彩!

    mask=cloud_mask,

    stopwords = STOPWORDS.add("資料"),       #屏蔽詞,屏蔽掉“資料”這個詞

    scale=10,

    max_words=2000,

    font_path="simsun.ttc",   #此處需要python環境中有對應TXT檔案中讀取的字型;

    min_font_size=6,

    max_font_size=40, 

    width=600,

    height=300

    )

wc.generate(bigdata)

wc.to_file("bigdata9.png")

輸出圖形

WordCloud中屏蔽詞+背景色設定

繼續閱讀