【实战教程】借助 CodeBuddy 轻松打造精美中文词云图 - AI 编程助手使用体验

from wordcloud import WordCloud
import matplotlib.pyplot as plt
import os
import jieba  # 中文分词库
from collections import Counter  # 词频统计

# 文件路径配置
input_file = 'text.txt'  # 输入文本文件
output_file = 'wordcloud.png'  # 输出图片文件
font_path = '字体家AI造字春风.ttf'  # 中文字体文件

# 检查字体文件是否存在
if not os.path.exists(font_path):
    print(f"警告：未找到中文字体文件 {font_path}")
    print("请下载字体家AI造字春风.ttf字体文件并放在当前目录")
    print("或修改代码中的font_path变量为你的中文字体路径")
    font_path = None  # 尝试使用系统默认字体

# 检查输入文件
if not os.path.exists(input_file):
    print(f"错误：当前目录下找不到 {input_file} 文件")
    print("请确保有一个名为 text.txt 的文本文件在当前目录下")
    exit()

# 读取文本内容
try:
    with open(input_file, 'r', encoding='utf-8') as file:
        text = file.read()
except Exception as e:
    print(f"读取文件时出错: {e}")
    exit()

# 中文分词处理函数
def chinese_text_segment(text):
    # 使用jieba进行中文分词
    seg_list = jieba.cut(text)
    # 过滤掉单个字符和空白（保留多字词语）
    words = [word for word in seg_list if len(word.strip()) > 1]
    # 统计词频
    word_counts = Counter(words)
    return word_counts

# 创建词云对象（自定义参数）
wordcloud = WordCloud(
    font_path=font_path,    # 中文字体路径（关键参数）
    width=800,              # 宽度
    height=600,             # 高度
    background_color='white',  # 背景色
    max_words=200,          # 最大词数
    max_font_size=150,      # 最大字体大小
    random_state=42,        # 随机种子（保证每次生成结果一致）
    collocations=False,     # 是否包含两个词的搭配
    stopwords=None,         # 停用词列表（可自定义）
    prefer_horizontal=0.9,  # 水平显示比例
    margin=2,              # 边距
    scale=2                # 缩放比例（提高清晰度）
)

# 生成词云
try:
    # 对中文文本进行分词和词频统计
    word_counts = chinese_text_segment(text)
    # 根据词频生成词云
    wordcloud.generate_from_frequencies(word_counts)
except Exception as e:
    print(f"生成词云时出错: {e}")
    exit()

# 显示词云
plt.figure(figsize=(10, 8))  # 设置图像大小
plt.imshow(wordcloud, interpolation='bilinear')  # 显示词云
plt.axis('off')  # 关闭坐标轴

# 保存词云图
try:
    wordcloud.to_file(output_file)
    print(f"词云图已成功保存为 {output_file}")
except Exception as e:
    print(f"保存词云图时出错: {e}")

# 显示图像
plt.show()