返回

在爱的节日里，用 Python 绘制美丽的词云

后端

七夕佳节，浪漫来袭。值此良辰，何不借用 Python，为心爱的 TA 绘制一幅别出心裁的词云图，传达浓浓爱意？

在本文中，我们将指导你一步步使用 Python 从聊天记录中提取关键词，并用 Matplotlib 绘制出美轮美奂的词云图，让你在七夕节给 TA 送上一份独一无二的惊喜。

一、打开聊天记录文件

首先，我们需要打开聊天记录文件。这里我们使用的是 QQ 聊天记录，你也可以根据实际情况选择其他聊天工具的记录文件。

import os

# 聊天记录文件路径
chat_file = "chat.txt"

# 打开聊天记录文件
with open(chat_file, "r", encoding="utf-8") as f:
    chat_text = f.read()

二、分词与词频统计

接下来，我们需要对聊天记录进行分词并统计词频。

import jieba

# 分词
words = jieba.cut(chat_text)

# 统计词频
word_freq = {}
for word in words:
    if word not in word_freq:
        word_freq[word] = 0
    word_freq[word] += 1

三、绘制词云

现在，我们可以使用 Matplotlib 绘制词云图了。

import matplotlib.pyplot as plt
from wordcloud import WordCloud

# 词云配置
wordcloud = WordCloud(
    background_color="white",
    width=800,
    height=600,
    max_words=200,
)

# 生成词云
wordcloud.generate_from_frequencies(word_freq)

# 显示词云
plt.imshow(wordcloud)
plt.axis("off")
plt.show()

四、保存词云图

最后，我们可以将词云图保存为图片。

# 保存词云图
wordcloud.to_file("wordcloud.png")

结语

恭喜你！你已经用 Python 绘制了一幅美丽的词云图。在七夕这个特殊的日子里，将这幅词云图分享给心爱的 TA，表达你浓浓的爱意。愿你们的爱情如词云图般绚烂多彩，甜甜蜜蜜。