利用python扩展包生成词云——wordcloud

环境

  • 系统 :macOS Catalina -V 10.15.4
  • IDE :Vscode -V 1.46.0
  • 语言 :Python -V 3.7.7
  • 库 :wordcloud、matplotlib、numpy、jieba、PIL、
    库的安装:

pip install wordcloud
pip install numpy
pip install matplotlib
pip install jieba

实例

结果图像:
利用python扩展包生成词云——wordcloud_第1张图片
模版图像:
利用python扩展包生成词云——wordcloud_第2张图片
代码及注解:

from wordcloud import WordCloud #词云
from matplotlib import pyplot as plt #图片绘制
from PIL import Image #图片处理
import numpy as np  #矩阵运算
import jieba #中文处理

with open('/Users/liuwentao/python/python基础/爬虫/活着.txt', 'r') as f:
    cont = f.read()
wcut = " ".join(jieba.cut(cont)) #利用jieba.cut()用于将字符串分割成一个一个的词汇,类型为generator,使用" ".jion()函数进行连接,同时类型转变为str

img = Image.open("/Users/liuwentao/python/python基础/爬虫/timg.jpg")
matrix = np.array(img) #图片对象被转换为数组,wordcloud会根据数组的位置显示词汇
wc = WordCloud(
    font_path = '/System/Library/Fonts/Supplemental/Songti.ttc', #指定字体类型,需要有字体文件
    mask = matrix, #指定掩版,也就是词云的显示框架
    max_words = 1000, #指定显示词汇的最大数量
    background_color = "white", #指定背景颜色,默认黑色
    random_state = 42
)
wc.generate_from_text(wcut)
plt.axis('off') #不显示坐标轴
plt.imshow(wc) #根据设定模版wc生成图像
# plt.show() #显示图像
plt.savefig('./活着.png', dpi = 300) #保存图像,指定路径和分辨率

参考资料:
wordcloud : http://amueller.github.io/word_cloud/auto_examples/wordcloud_cn.html

你可能感兴趣的:(python,github)