word_cloud-用Python之作个性化词云图

网上有很多制作词云的网站,我们使用Python也可以很方便的制作,这里,我们就简单学习下。

1. word_cloud

GitHub地址:https://github.com/amueller/word_cloud
首先我们需要安装,正常来说,直接就执行

pip install wordcloud

即可,但是,我这个是在Windows平台,安装的时候,提示什么少了,需要去下载个编译器的,报错信息后面有URL,


word_cloud-用Python之作个性化词云图_第1张图片

这里忘记记下来了,遇到的同学,直接去下载下应该就行了

error: Microsoft Visual C++ 14.0 is required

这里,我从网上找了另一个方法解决
去这个网站:http://www.lfd.uci.edu/~gohlke/pythonlibs/#wordcloud
找到指定的版本下载就行了

word_cloud-用Python之作个性化词云图_第2张图片

这里,我就下载了这个cp36的,然后直接安装就行了

pip install wordcloud-1.3.2-cp36-cp36m-win_amd64.whl

官网上有个小栗子,我们可以测试下,代码和使用到的数据,github上都有

from os import path
from wordcloud import WordCloud

d = path.dirname(__file__)

# Read the whole text.
text = open(path.join(d, 'constitution.txt')).read()

# Generate a word cloud image
wordcloud = WordCloud().generate(text)

# Display the generated image:
# the matplotlib way:
import matplotlib.pyplot as plt
plt.imshow(wordcloud, interpolation='bilinear')
plt.axis("off")

# lower max_font_size
wordcloud = WordCloud(max_font_size=40).generate(text)
plt.figure()
plt.imshow(wordcloud, interpolation="bilinear")
plt.axis("off")
plt.show()

# The pil way (if you don't have matplotlib)
# image = wordcloud.to_image()
# image.show()
word_cloud-用Python之作个性化词云图_第3张图片
我们通过wordcloud生成了一个图片,然后使用matplotlib将图片展示出来,我们分析下上面的代码
matplotlib.pyplot.imshow(X, cmap=None, norm=None, aspect=None, interpolation=None, alpha=None, vmin=None, vmax=None, origin=None, extent=None, shape=None, filternorm=1, filterrad=4.0, imlim=None, resample=None, url=None, hold=None, data=None, **kwargs)

Display an image on the axes.

class wordcloud.WordCloud(font_path=None, width=400, height=200, margin=2, ranks_only=None, prefer_horizontal=0.9, mask=None, scale=1, color_func=None, max_words=200, min_font_size=4, stopwords=None, random_state=None, background_color='black', max_font_size=None, font_step=1, mode='RGB', relative_scaling=0.5, regexp=None, collocations=True, colormap=None, normalize_plurals=True)

这里有很多例子:https://amueller.github.io/word_cloud/auto_examples/index.html

2. 中文词云

上面,我们试试中文,我们换一个中文的文件试试即可

东方网记者王佳妮8月19日报道:8月18日晚,一则重磅消息在沪发布:即日起,上海暂停新增投放共享单车。上海市交通委表示,将采取有效措施,逐步解决共享单车快速发展带来的无序和不平衡问题,促进共享单车行业的持续健康发展。东方网记者从各平台获悉,将积极配合,同时考虑将中心城区部分车辆向郊区转移。

word_cloud-用Python之作个性化词云图_第4张图片

执行后,发现,中文都是乱码
我们观察上面wordcloud类,他有一个参数

font_path : string

    Font path to the font that will be used (OTF or TTF). Defaults to DroidSansMono path on a Linux machine. If you are on another OS or don’t have this font, you need to adjust this path.

这个可以指定为中文字体,

font = r'C:\Windows\Fonts\simsun.ttc'
# Generate a word cloud image
wordcloud = WordCloud(font_path=font).generate(text)

再次执行,中文可以正常显示了


word_cloud-用Python之作个性化词云图_第5张图片

但是,显示的并不是我们想要的,
正常词云的话,显示的是关键词,和他的出现频率有关,这里的话,并没有对文本内容做很好的分词
刚刚去简单整理了一个Python的分词:Python中文分词-jieba

这样,我们继续修改下上面的代码

from os import path
from wordcloud import WordCloud
import jieba

d = path.dirname(__file__)

# Read the whole text.
text = open(path.join(d, 'hello.txt'),encoding='utf-8').read()

text_cut = jieba.cut(text , cut_all=False)
font = r'C:\Windows\Fonts\simsun.ttc'
# Generate a word cloud image
wordcloud = WordCloud(font_path=font).generate(' '.join(text_cut))

# Display the generated image:
# the matplotlib way:
import matplotlib.pyplot as plt
plt.imshow(wordcloud, interpolation='bilinear')
plt.axis("off")

再次执行下


word_cloud-用Python之作个性化词云图_第6张图片

好嘞,这下差不多了,我们的中文词云图就搞好了

3. 自定义形状的词云图

下面,我们看看自定义形状的词云图,这里用官方的例子,就是需要多传入一个mask图片就行了

# -*- coding: utf-8 -*-
"""
Created on Sun Aug 20 14:04:48 2017

@author: yuguiyang
"""


from os import path
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt

from wordcloud import WordCloud, STOPWORDS

d = path.dirname(__file__)

# Read the whole text.
text = open(path.join(d, 'alice.txt'),encoding='utf-8').read()

# read the mask image
# taken from
# http://www.stencilry.org/stencils/movies/alice%20in%20wonderland/255fk.jpg
alice_mask = np.array(Image.open(path.join(d, "alice_mask.png")))

stopwords = set(STOPWORDS)
stopwords.add("said")

wc = WordCloud(background_color="white", max_words=8000, mask=alice_mask,
               stopwords=stopwords)
# generate word cloud
wc.generate(text)

# store to file
wc.to_file(path.join(d, "alice.png"))

# show
plt.imshow(wc, interpolation='bilinear')
plt.axis("off")
plt.figure()
plt.imshow(alice_mask, cmap=plt.cm.gray, interpolation='bilinear')
plt.axis("off")
plt.show()
word_cloud-用Python之作个性化词云图_第7张图片

刚试了下,就是要找一个喜欢的图形图片,


word_cloud-用Python之作个性化词云图_第8张图片

你可能感兴趣的:(word_cloud-用Python之作个性化词云图)