python生成词云图_python生成词云图

#!/usr/bin/python

# coding: UTF-8

# coding:utf8

# wordcloud库默认为英文文本设置，默认字体也是英文

# jieba支持中文

import sys

from os import path

import os

from PIL import Image

import numpy as np

import matplotlib.pyplot

import jieba

import jieba.analyse

from wordcloud import WordCloud

matplotlib.use('TkAgg')

d=path.dirname(__file__)

# stopwords_path = 'stopwords\stopwords1893.txt' # 停用词词表

# 添加的自定义中文语句的代码在这里

jieba.add_word('路明非')

# 读取整个文本-要分析的文本

text=open(path.join(d,'doubancomment.csv')).read()

def jiebaclearText(text):

mywordlist = []

seg_list = jieba.cut(text, cut_all=False) #精确模式

liststr="/ ".join(seg_list)

f_stop = open(path.join(d,'stopwords1893.txt'))

try:

f_stop_text = f_stop.read( )

f_stop_text= f_stop_text.encode("utf-8")

finally:

f_stop.close( )

f_stop_seg_list=str(f_stop_text,encoding="utf-8").split('\n')

for myword in liststr.split('/'):

if not(myword.strip() in f_stop_seg_list) and len(myword.strip())>1:

mywordlist.append(myword)

return ''.join(mywordlist)

text1 = jiebaclearText(text)

print(text1)

file_path = os.path.abspath('.')

alice_mask=np.array(Image.open(path.join(d,"pyworld.png")))

font =file_path+ '/huawen.ttf' #一定要设置中文字体否则显示乱码 先下载

print(file_path)

wc=WordCloud(background_color="black",font_path=font,max_words=2000,mask=alice_mask,stopwords=open(path.join(d,file_path + "/stopwords/stopwords1893.txt")).read())

# 生成一个词云图像

wordcloud=wc.generate(text1)

image=wordcloud.to_image()

# 展示生成的词云图像

image.show()