Python使用词云图展示

网上看到一个txt文本信息,共2351条饭否记录,据说是微信之父每天发的饭否记录,其实我不知道什么是饭否。我读取这个文本内容,展示到词语图上。之前也使用过,但是好久没有玩Python了,称假期空闲,练习练习。开始发行“通过网页”变成了高频词汇,一看源文本文件,发行每条记录的后面都包含“ 2010-11-26 13:59 通过网页”,这样通过网页肯定是高频词了,所有重新处理了源文本信息,使用正则表达式式,提前时间节点前的任意字符。使用的正则表达式:.+(?=(\d{4}\-\d{2}\-\d{2}))

提取信息如下:

Python使用词云图展示_第1张图片

找一个背景,网上找奋斗图片,去掉背景色,裁剪人物。

Python使用词云图展示_第2张图片

完整代码:

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

from wordcloud import WordCloud, STOPWORDS

# get data directory (using getcwd() is needed to support running example in generated IPython notebook)
d = path.dirname(__file__) if "__file__" in locals() else os.getcwd()
print(d)

# Read the whole text.
# text = open(path.join(d, 'ZXL.txt'), encoding='utf-8', errors='ignore').read()
with open(path.join(d, 'ZXL.txt'), 'rb') as f:
    raw_data = f.read()
    result = chardet.detect(raw_data)
    use_encoding = result['encoding']

# 查看文本使用的编码
print(use_encoding) # utf-8
text = open(path.join(d, 'ZXL.txt'), 'r', encoding=use_encoding).read()

# read the mask image
fight_mask = np.array(Image.open(path.join(d, "fight.png")))

# 指定字体文件路径
font_path = r'C:\Windows\Fonts\方正粗黑宋简体.ttf'

# 创建词云图对象并设置字体
stopwords = set(STOPWORDS)

wc = WordCloud(background_color="white",
               max_words=6000, mask=fight_mask,
               font_path=font_path,
               stopwords=stopwords,
               contour_width=3,
               contour_color='steelblue')

# generate word cloud
wc.generate(text)

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

# show
plt.imshow(wc, interpolation='bilinear')
plt.axis("off")
plt.show()


运行效果如下:

Python使用词云图展示_第3张图片

生产图片文件:

Python使用词云图展示_第4张图片

 

你可能感兴趣的:(python,开发语言,中文分词)