Update script_2.2.2.py

script_2.2.2.py

@@ -1,41 +1,41 @@
-import jieba
+import jieba
-import wordcloud
+import wordcloud
-import imageio
+import imageio
-import pandas as pd  # 用于读取Excel文件
+import pandas as pd  # 用于读取Excel文件
-
+
-# 读取图片作为词云的形状
+# 读取图片作为词云的形状
-img = imageio.imread('四叶草1.png')
+img = imageio.imread('四叶草1.png')
-
+
-# 使用pandas读取Excel文件中的所有列
+# 使用pandas读取Excel文件中的所有列
-df = pd.read_excel('top_ai_danmakus.xlsx')
+df = pd.read_excel('top_ai_danmakus.xlsx')
-
+
-# 将Excel所有列的数据拼接成一个长字符串
+# 将Excel所有列的数据拼接成一个长字符串
-book = ''
+book = ''
-
+
-# 遍历每一列并将内容合并成字符串
+# 遍历每一列并将内容合并成字符串
-for column in df.columns:
+for column in df.columns:
-    # 将当前列转换为字符串并拼接到 book 变量
+    # 将当前列转换为字符串并拼接到 book 变量
-    book += ' '.join(df[column].astype(str)) + ' '
+    book += ' '.join(df[column].astype(str)) + ' '
-
+
-# 使用jieba进行分词
+# 使用jieba进行分词
-book_list = jieba.lcut(book)
+book_list = jieba.lcut(book)
-book_str = ' '.join(book_list)
+book_str = ' '.join(book_list)
-
+
-# 打印分词后的字符串
+# 打印分词后的字符串
-print(book_str)
+print(book_str)
-
+
-# 创建词云
+# 创建词云
-wc = wordcloud.WordCloud(
+wc = wordcloud.WordCloud(
-    width=500,
+    width=500,
-    height=500,
+    height=500,
-    background_color='white',
+    background_color='white',
-    mask=img,
+    mask=img,
-     stopwords={'main','Taipei','afraid','aiden','Britain'},  # 可添加不需要的停用词
+     stopwords={'main','Taipei','afraid','aiden','Britain'},  # 添加停用词
-    font_path='msyh.ttc'  # 设置中文字体路径
+    font_path='msyh.ttc'  # 设置中文字体路径
-)
+)
-
+
-# 生成词云
+# 生成词云
-wc.generate(book_str)
+wc.generate(book_str)
-
+
-# 保存词云图
+# 保存词云图
-wc.to_file('词云图.png')
+wc.to_file('词云图.png')