From e56f05116d5407b6547c87b1c79f5ff41158d8b4 Mon Sep 17 00:00:00 2001
From: pufahrcyp <1195744232@qq.com>
Date: Wed, 18 Sep 2024 22:07:19 +0800
Subject: [PATCH] =?UTF-8?q?ADD=20=E7=94=9F=E6=88=90=E8=AF=8D=E4=BA=91?=
 =?UTF-8?q?=E5=9B=BE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 word_cloud.py | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)
 create mode 100644 word_cloud.py

diff --git a/word_cloud.py b/word_cloud.py
new file mode 100644
index 0000000..1bc121d
--- /dev/null
+++ b/word_cloud.py
@@ -0,0 +1,35 @@
+import jieba
+import wordcloud
+import imageio
+# 导入imageio库中的imread函数，并用这个函数读取本地图片，作为词云形状图片
+
+
+def generate(file_path):
+    print("正在生成词云图...")
+    py = imageio.imread('.\\03.png')  # 巴黎铁塔的蒙版图
+    # 读取文件内容
+    f = open(file_path, encoding='utf-8')
+    txt = f.read()
+    # jiabe 分词 分割词汇
+    txt_list = jieba.lcut(txt)
+    string = ' '.join(txt_list)
+    # 词云图设置
+    wc = wordcloud.WordCloud(
+        width=1000,         # 图片的宽
+        height=700,         # 图片的高
+        background_color='white',   # 图片背景颜色
+        font_path='msyh.ttc',    # 词云字体
+        mask=py,     # 所使用的词云图片
+        scale=15,
+        # contour_color = 'steelblue' ,
+        contour_color='gold',  # 设置轮廓为金色 (使用颜色名称)
+        contour_width=15,       # 轮廓宽度
+        # 停用词
+        stopwords={'是', '的', '和', '吗', '这是', '不是', '了', '还', '这个', '哈', '啊'}
+        )
+    # 给词云输入文字
+    wc.generate(string)
+    search_word = file_path.rsplit(".", 1)[0]
+    # 保存词云图
+    wc.to_file(f'{search_word}词云图.png')
+    print("成功保存词云图！")
\ No newline at end of file