当前位置：首页 > article >正文

Python数据分析可视化之词云图

article 2024/12/25 22:19:39

要使用自定义形状和颜色映射生成中文词云图并保存到指定位置，你需要按照以下步骤操作。我们将结合 wordcloud、matplotlib、jieba 以及 PIL（Python Imaging Library）来完成这个任务。

步骤 1：安装必要的库

确保你已经安装了以下库：

pip install wordcloud matplotlib jieba pillow

wordcloud：用于生成词云图。
matplotlib：用于显示图像。
jieba：用于中文分词。
Pillow：用于处理图像文件。

步骤 2：准备文本数据

假设你有一个包含中文文本的文件 chinese_text.txt，我们将从该文件中读取文本数据。

步骤 3：准备形状模板

你需要一个 PNG 图像作为词云的形状模板。例如，你可以使用一个心形或圆形的 PNG 文件。确保图像为白色背景，黑色填充（即非零像素将作为词云的形状）。我们将使用 PIL 来读取这个图像文件。

步骤 4：编写代码

以下是完整的代码示例，展示如何使用自定义形状和颜色映射生成中文词云图，并将其保存到指定位置。

示例代码

from wordcloud import WordCloud, STOPWORDS
import matplotlib.pyplot as plt
import jieba
import numpy as np
from PIL import Image
import os

# 1. 读取文本文件
def read_file(file_path):
    with open(file_path, 'r', encoding='utf-8') as f:
        return f.read()

# 2. 使用 jieba 进行中文分词
def preprocess_text(text):
    # 分词
    words = jieba.lcut(text)
    # 去除停用词
    stopwords = set(STOPWORDS) | set(['的', '是', '在', '和', '我', '这', '它', '你', '有', '一', '不', '了', '都', '而', '与'])
    filtered_words = [word for word in words if word not in stopwords and len(word.strip()) > 0]
    return " ".join(filtered_words)

# 3. 读取形状模板图像
def get_mask_image(mask_path):
    mask = np.array(Image.open(mask_path))
    return mask

# 4. 生成词云图
def generate_wordcloud(text, mask, output_path, font_path, colormap='viridis'):
    # 创建词云对象
    wordcloud = WordCloud(
        width=800,
        height=400,
        background_color='white',
        font_path=font_path,  # 指定支持中文的字体路径
        max_words=100,
        mask=mask,  # 使用自定义形状
        colormap=colormap,  # 设置颜色映射
        stopwords=STOPWORDS  # 使用默认停用词列表
    ).generate(text)

    # 保存词云图到指定位置
    wordcloud.to_file(output_path)

    # 显示词云图
    plt.figure(figsize=(10, 5))
    plt.imshow(wordcloud, interpolation='bilinear')
    plt.axis('off')  # 关闭坐标轴
    plt.show()

# 5. 主函数
def main():
    # 文件路径
    text_file_path = 'path/to/chinese_text.txt'  # 替换为你的文本文件路径
    mask_image_path = 'path/to/mask.png'  # 替换为你的形状模板图像路径
    output_image_path = 'path/to/output/wordcloud.png'  # 替换为你想要保存的路径
    font_path = 'path/to/simhei.ttf'  # 替换为你的中文字体文件路径

    # 确保输出目录存在
    output_dir = os.path.dirname(output_image_path)
    if not os.path.exists(output_dir):
        os.makedirs(output_dir)

    # 读取文本
    text = read_file(text_file_path)

    # 预处理文本
    processed_text = preprocess_text(text)

    # 读取形状模板
    mask = get_mask_image(mask_image_path)

    # 生成并保存词云图
    generate_wordcloud(processed_text, mask, output_image_path, font_path, colormap='plasma')

if __name__ == '__main__':
    main()

代码说明

read_file(file_path)：读取文本文件的内容。
preprocess_text(text)：使用 jieba 对中文文本进行分词，并去除常见的停用词。你可以根据需要调整停用词列表。
get_mask_image(mask_path)：读取形状模板图像，并将其转换为 NumPy 数组，以便 wordcloud 可以使用。
generate_wordcloud(text, mask, output_path, font_path, colormap)：创建词云对象，生成词云图，并将其保存到指定位置。我们还使用了 matplotlib 来显示生成的词云图。
main()：主函数，负责调用上述所有功能。你需要替换文件路径为实际的路径。

自定义选项

colormap：你可以选择不同的颜色映射。matplotlib 提供了许多内置的颜色映射，如 'viridis', 'plasma', 'inferno', 'magma' 等。你可以通过 colormap 参数指定你想要的颜色映射。
font_path：确保你指定了一个支持中文的字体文件路径（如 simhei.ttf）。你可以从网上下载适合的字体文件，或者使用系统自带的字体。
mask：传入一个 PNG 图像作为词云的形状模板。图像中的非零像素将作为词云的形状。你可以使用任何图像作为模板，例如圆形、心形等。