首页 > 代码库 > 使用python绘制词云

使用python绘制词云

2024-08-25 01:47:19   217人阅读

最近在忙考试的事情，没什么时间敲代码，一个月也没几天看代码，最近看到可视化的词云，看到网上也很多这样的工具，

但是都不怎么完美，有些不支持中文，有的中文词频统计得莫名其妙、有的不支持自定义形状、所有的都不能自定义颜色

于是网上找了一下，决定用python绘制词云，主要用到的是wordcloud库，安装只需要pip isntall wordcloud就行，

数据用的是酒店评论的数据，代码如下：

# -*- coding: utf-8 -*-
import matplotlib.pyplot as plt
import pickle
from wordcloud import WordCloud,STOPWORDS,ImageColorGenerator
import jieba
# import codecs

# fin = codecs.open(‘HotelComments.txt‘,mode = ‘r‘, encoding = ‘utf-8‘)
# print fin.read()

# 第一次运行程序时将分好的词存入文件
# text = ‘‘
# with open(‘HotelComments.txt‘) as fin:
#     for line in fin.readlines():
#         line = line.strip(‘\n‘)
#         text += ‘ ‘.join(jieba.cut(line))
#         text += ‘ ‘
# fout = open(‘text.txt‘,‘wb‘)
# pickle.dump(text,fout)
# fout.close()

# 直接从文件读取数据
fr = open(‘text.txt‘,‘rb‘)
text = pickle.load(fr)

backgroud_Image = plt.imread(‘girl.jpg‘)
wc = WordCloud( background_color = ‘white‘,    # 设置背景颜色
                mask = backgroud_Image,        # 设置背景图片
                max_words = 2000,            # 设置最大现实的字数
                stopwords = STOPWORDS,        # 设置停用词
                font_path = ‘C:/Users/Windows/fonts/msyh.ttf‘,# 设置字体格式，如不设置显示不了中文
                max_font_size = 50,            # 设置字体最大值
                random_state = 30,            # 设置有多少种随机生成状态，即有多少种配色方案
                )
wc.generate(text)
image_colors = ImageColorGenerator(backgroud_Image)
wc.recolor(color_func = image_colors)
plt.imshow(wc)
plt.axis(‘off‘)
plt.show()

效果图：

技术分享

自定义词云形状：

技术分享

 

参考博客：

http://blog.csdn.net/tanzuozhev/article/details/50789226

http://blog.csdn.net/qq_16912257/article/details/52458515

使用python绘制词云


联系
我们