微信大众号:愤恨的it男,超多Python技能干货文章。

一、简单介绍一下

云图是文本发掘中用来表征词频的数据可视化图画,经过它能够很直观地展现文本数据中地高频词,让读者能够从大量文本数据中快速抓住要点。如下图:

wordcloud,一个超酷的python库

wordcloud则是一个十分优异的词云展现python库,它支持自定义词云图的巨细、色彩、字体等,甚至能够经过蒙版图片设置词云图的形状。因此,咱们能够凭借wordcloud轻松生成精巧的词云图。

二、装置只需一行命令

pip install wordcloud

三、从一个简单比如开端

from wordcloud import WordCloud
text = "微信大众号:愤恨的it男"
wc = WordCloud(font_path='FZYTK.TTF', repeat=True)
wc.generate(text)
wc.to_file('wordcloud.png')

这里经过WordCloud类设置字体为方正姚体,布景色彩为白色,文本能够重复显示。生成WordCloud目标后,运用generate()办法将“微信大众号:愤恨的it男”生成词云图。最后,运用to_file()办法生成图片文件。

wordcloud,一个超酷的python库

四、细说wordcloud

WordCloud作为wordcloud库最核心的类,其主要参数及阐明如下:

wordcloud,一个超酷的python库

这里以wordcloud库官方文档的constitution.txt文件作为数据,掩盖WordCloud类的各种参数设置用法,制作出一张精巧的词云图。

wordcloud,一个超酷的python库

首先,读入constitution.txt数据,并将数据清洗成空格分隔的长字符串。

import re
with open('constitution.txt') as c:
    text = ' '.join([word.group().lower() for word in re.finditer('[a-zA-Z] ', c.read())])
print(text[:500])

wordcloud,一个超酷的python库

然后,在默许参数设置下,运用WordCloud目标的generate()和to_file()办法生成一张简单的词云图。

from wordcloud import WordCloud
import re
with open('constitution.txt') as c:
    text = ' '.join([word.group().lower() for word in re.finditer('[a-zA-Z] ', c.read())])
wc = WordCloud()
wc.generate(text)
wc.to_file('wordcloud.png')

wordcloud,一个超酷的python库

以上词云图是在默许参数下生成的,简单粗糙不好看。接下来咱们将对WordCloud的各种参数调整设置,不断地对以上词云图进行晋级改造。

1、设置图片特点

设置图片宽为600,高为300,扩大1.5倍,色彩空间为RGBA,布景色彩为None。

from wordcloud import WordCloud
import re
with open('constitution.txt') as c:
    text = ' '.join([word.group().lower() for word in re.finditer('[a-zA-Z] ', c.read())])
wc = WordCloud(
    width=600,
    height=300,
    scale=1.5,
    mode='RGBA',
    background_color=None,
    )
wc.generate(text)
wc.to_file('wordcloud.png')

wordcloud,一个超酷的python库

2、设置文字布局

设置水平比例为1(即全部为水平文字),最多只显示100个词,停用词运用自带的词典(中文需要传入自定义的),相关一致性为0.3,文字布局为非随机,不允许重复词。

from wordcloud import WordCloud
import re
with open('constitution.txt') as c:
    text = ' '.join([word.group().lower() for word in re.finditer('[a-zA-Z] ', c.read())])
wc = WordCloud(
    width=600,
    height=300,
    scale=1.5,
    mode='RGBA',
    background_color=None,
    prefer_horizontal=1,
    max_words=400,
    stopwords=None,
    relative_scaling=0.3,
    random_state=4,
    repeat=False,
    )
wc.generate(text)
wc.to_file('wordcloud.png')

wordcloud,一个超酷的python库

3、设置字体特点

设置字体为‘JOKERMAN.TTF’,最小字号为2,最大字号为150。

from wordcloud import WordCloud
import re
with open('constitution.txt') as c:
    text = ' '.join([word.group().lower() for word in re.finditer('[a-zA-Z] ', c.read())])
wc = WordCloud(
    width=600,
    height=300,
    scale=1.5,
    mode='RGBA',
    background_color=None,
    prefer_horizontal=1,
    max_words=400,
    stopwords=None,
    relative_scaling=0.3,
    random_state=4,
    repeat=False,
    font_path='JOKERMAN.TTF',
    min_font_size=2,
    max_font_size=150,
    )
wc.generate(text)
wc.to_file('wordcloud.png')

wordcloud,一个超酷的python库

4、设置蒙版

wordcloud,一个超酷的python库

设置微信大众号【愤恨的it男】头像的黑白图片为蒙版图片。

from PIL import Image
from wordcloud import WordCloud
import numpy as np
import re
mask_picture = np.array(Image.open('angry_it_man_mask.png'))
with open('constitution.txt') as c:
    text = ' '.join([word.group().lower() for word in re.finditer('[a-zA-Z] ', c.read())])
wc = WordCloud(
    width=600,
    height=300,
    scale=1.5,
    mode='RGBA',
    background_color=None,
    prefer_horizontal=1,
    max_words=400,
    stopwords=None,
    relative_scaling=0.3,
    random_state=4,
    repeat=False,
    font_path='JOKERMAN.TTF',
    min_font_size=2,
    max_font_size=150,
    mask=mask_picture,
    )
wc.generate(text)
wc.to_file('wordcloud.png')

wordcloud,一个超酷的python库

微信大众号:愤恨的it男,超多Python技能干货文章。