diff --git a/get_wordcloud.py b/get_wordcloud.py new file mode 100644 index 0000000..d8de295 --- /dev/null +++ b/get_wordcloud.py @@ -0,0 +1,33 @@ +import jieba +import wordcloud +import imageio + +f = open("danmu.txt",encoding = 'utf-8') +text = f.read() +text_list = jieba.lcut(text) +text_str = ' '.join(text_list) + +img = imageio.imread('五环.png') +# 制作方形词云图 +wc1 = wordcloud.WordCloud( + width=500, # 设置宽度 + height=500, # 设置高度 + background_color='white', # 设置背景颜色 + stopwords={'?', '了', '的', '我', '吗', '是', '这', '都', '啊', '也', '不', '他', '她', '人', '要', '又', '你'},# 屏蔽掉到一些没有意义的词 + font_path='msyh.ttc', # 设置字体文件 +) +wc1.generate(text_str) +wc1.to_file('WordCloud_1.png') +# 制作五环形状词云图 +wc2 = wordcloud.WordCloud( + width = 500, # 设置宽度 + height = 500, # 设置高度 + background_color = 'white', # 设置背景颜色 + mask = img, # 设置形状 + stopwords = {'?','了','的','我','吗','是','这','都','啊','也','不','他','她','人','要','又','你'}, # 屏蔽掉到一些没有意义的词 + font_path = 'msyh.ttc', # 设置字体文件 +) +wc2.generate(text_str) +wc2.to_file('WordCloud_2.png') + +