天天看点

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中屏蔽词+背景色设置

继续阅读