一、环境搭建
使用anaconda建立科学计算环境,并安装相关软件包
root@test:~/python/wordcloud# conda info -e
# conda environments:
#
science /usr/local/miniconda2/envs/science
science_init /usr/local/miniconda2/envs/science_init
root * /usr/local/miniconda2
root@test:~/python/wordcloud# source activate science
#安装相关包
(science) root@test:~/python/wordcloud# pip install jieba wordcloud
(science) root@test:~/python/wordcloud# conda install numpy PIL matplotlib
二、脚本
#!/usr/local/miniconda2/envs/science/bin/python
#-*- coding: utf-8 -*-
from os import path
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt
import jieba
from wordcloud import WordCloud
#读取脚本所在目录
mulu = path.dirname(__file__)
#从文本中生成关键词
text = open(path.join(mulu, 'wordcloud.txt')).read()
seg_list = jieba.cut(text, cut_all=True)
seg_split = " ".join(seg_list)
#读取图片
wordcloud_mask = np.array(Image.open(path.join(mulu, 'timg.jpg')))
#字体位置
font = path.join(mulu, 'simsun.ttc')
#设置图片属性
wc = WordCloud(font_path=font, background_color="white", max_words=2000, mask=wordcloud_mask)
wc.generate(seg_split)
#生成新的图片
wc.to_file(path.join(mulu, "timg_new.jpg"))
#图片展示
plt.imshow(wc, interpolation='bilinear')
plt.axis("off")
plt.show()