diff --git a/WordCloud.py b/WordCloud.py index d037451..69b5080 100644 --- a/WordCloud.py +++ b/WordCloud.py @@ -1,40 +1,38 @@ -import matplotlib.pyplot as plt -import wordcloud -import jieba -import pyecharts -import pandas as pd -import numpy as np -from PIL import Image -import nltk -from nltk import FreqDist - -font_path = r"C:\Users\26716\AppData\Local\Microsoft\Windows\Fonts\HanYiDieYuTiJian-1.ttf" - -with open("danmu.txt", 'r', encoding='utf-8') as f: - danmu_text = f.read() - -stopword = pd.read_csv("stopwords.txt", names=["w"], sep="\t", encoding="utf-8") - -jieba.add_word("发来贺电") -words = jieba.cut(danmu_text) - -filtered_words = [w for w in words if w not in stopword['w'].tolist() and len(w) > 1] - -top_words = FreqDist(filtered_words) - -background = np.array(Image.open("D:\CODING\软工\奥运五环.png").convert("RGB")) - -cloudobj =wordcloud.WordCloud(font_path = font_path, - scale = 4, - width = 1920, - height= 1080, - max_words= 100, - background_color= "white", - max_font_size= 200, - min_font_size= 10, - mask = background - ).fit_words(top_words) - -plt.imshow(cloudobj) -plt.axis("off") -plt.show() +import matplotlib.pyplot as plt +import wordcloud +import jieba +import pandas as pd +import numpy as np +from PIL import Image +from nltk import FreqDist + +font_path = r"C:\Users\26716\AppData\Local\Microsoft\Windows\Fonts\HanYiDieYuTiJian-1.ttf" + +with open("danmu.txt", 'r', encoding='utf-8') as f: + danmu_text = f.read() + +stopword = pd.read_csv("stopwords.txt", names=["w"], sep="\t", encoding="utf-8") + +jieba.add_word("发来贺电") +words = jieba.cut(danmu_text) + +filtered_words = [w for w in words if w not in stopword['w'].tolist() and len(w) > 1] + +top_words = FreqDist(filtered_words) + +background = np.array(Image.open("D:\CODING\软工\奥运五环.png").convert("RGB")) + +cloudobj =wordcloud.WordCloud(font_path = font_path, + scale = 4, + width = 1920, + height= 1080, + max_words= 100, + background_color= "white", + max_font_size= 200, + min_font_size= 10, + mask = background + ).fit_words(top_words) + +plt.imshow(cloudobj) +plt.axis("off") +plt.show()