1 import jieba 2 from matplotlib import pyplot as plt 3 from wordcloud import WordCloud 4 from PIL import Image 5 import numpy as np 6 txt = (open("红楼梦.txt", "r", encoding='utf-8')).read() 7 file1 = open("stopwords_cn.txt") 8 file2 = open("stopwords_cn(more).txt") 9 ls1 = [] 10 while 1: 11 line = file1.readline() 12 new_word = line.strip() 13 if not line: 14 break 15 ls1.append(new_word) 16 ls2 = [] 17 while 1: 18 line = file2.readline() 19 new_word = line.strip() 20 if not line: 21 break 22 ls2.append(new_word) 23 ls = ls1+ls2 24 words = jieba.lcut(txt) 25 counts = {} 26 for word in words: 27 for i in ls: 28 if word == i: 29 continue 30 if (len(word)) == 1: 31 continue 32 else: 33 counts[word] = counts.get(word, 0) + 1 34 items = list(counts.items()) 35 items.sort(key=lambda x: x[1], reverse=True) 36 for i in range(15): 37 word, count = items[i] 38 print("{0:<10}{1:>5}".format(word, count)) 39 string = ' '.join(words) 40 print(len(string)) 41 img = Image.open('22.png') #打开图片 42 img_array = np.array(img) #将图片装换为数组 43 stopword=['什么', '一个', '我们', '那里', '你们', '如今', '起来', '知道', '这里', '众人', '他们', '出来', '自己', '说道', '听见', '两个', '姑娘', '不好', 44 '不知', '只见', '东西', '告诉'] #设置停止词，也就是你不想显示的词，这里这个词是我前期处理没处理好，你可以删掉他看看他的作用 45 stopword=stopword+ls 46 print(stopword) 47 wc = WordCloud( 48 background_color='white', 49 width=1000, 50 height=800, 51 mask=img_array, 52 font_path='./fonts/simhei.ttf', 53 stopwords=stopword 54 ) 55 wc.generate_from_text(string)#绘制图片 56 plt.imshow(wc) 57 plt.axis('off') 58 plt.figure() 59 plt.show() #显示图片 60 wc.to_file('new.png') #保存图片
