生成词云图

test3.py

@@ -0,0 +1,34 @@
+import pandas as pd
+from wordcloud import WordCloud, ImageColorGenerator
+import matplotlib.pyplot as plt
+from PIL import Image
+import numpy as np
+
+# 文件路径
+input_file = 'danmu.txt'  # 输入文件名
+
+# 从文本文件中读取弹幕数据
+with open(input_file, 'r', encoding='utf-8') as file:
+    danmaku_text = file.read()
+
+# 创建词云对象
+wordcloud = WordCloud(
+    width=800,
+    height=600,
+    background_color="white",
+    max_words=200,
+    colormap='viridis',  # 颜色主题
+    random_state=42,
+    font_path='C:\Windows\Fonts\STZHONGS.TTF'  # 字体路径，如果需要支持中文显示
+).generate(danmaku_text)
+
+# 可选：使用自定义图片作为背景
+mask = np.array(Image.open('image.jpg'))  # 替换为你的图片文件名
+image_colors = ImageColorGenerator(mask)
+wordcloud.recolor(color_func=image_colors)
+
+plt.figure(figsize=(10, 8))
+plt.imshow(wordcloud, interpolation='bilinear')
+plt.axis("off")  # 不显示坐标轴
+plt.title('弹幕词云图')
+plt.show()