Python词云分析

 import jieba
 from matplotlib import pyplot as plt
 from wordcloud import WordCloud
 from PIL import Image
 import numpy as np
 txt = (open("红楼梦.txt", "r", encoding='utf-8')).read()
 file1 = open("stopwords_cn.txt")
 file2 = open("stopwords_cn(more).txt")
 ls1 = []
 while 1:
     line = file1.readline()
     new_word = line.strip()
     if not line:
         break
     ls1.append(new_word)
 ls2 = []
 while 1:
     line = file2.readline()
     new_word = line.strip()
     if not line:
         break
     ls2.append(new_word)
 ls = ls1+ls2
 words = jieba.lcut(txt)
 counts = {}
 for word in words:
     for i in ls:
         if word == i:
             continue
     if (len(word)) == 1:
         continue
     else:
         counts[word] = counts.get(word, 0) + 1
 items = list(counts.items())
 items.sort(key=lambda x: x[1], reverse=True)
 for i in range(15):
     word, count = items[i]
     print("{0:<10}{1:>5}".format(word, count))
 string = ' '.join(words)
 print(len(string))
 img = Image.open('22.png') #打开图片
 img_array = np.array(img) #将图片装换为数组
 stopword=['什么', '一个', '我们', '那里', '你们', '如今', '起来', '知道', '这里', '众人', '他们', '出来', '自己', '说道', '听见', '两个', '姑娘', '不好',
           '不知', '只见', '东西', '告诉']  #设置停止词，也就是你不想显示的词，这里这个词是我前期处理没处理好，你可以删掉他看看他的作用
 stopword=stopword+ls
 print(stopword)
 wc = WordCloud(
     background_color='white',
     width=1000,
     height=800,
     mask=img_array,
     font_path='./fonts/simhei.ttf',
     stopwords=stopword
 )
 wc.generate_from_text(string)#绘制图片
 plt.imshow(wc)
 plt.axis('off')
 plt.figure()
 plt.show()  #显示图片
 wc.to_file('new.png')  #保存图片