基于DataFrame结构的词频表生成词云图

最新推荐文章于 2024-09-06 10:00:00 发布

原创最新推荐文章于 2024-09-06 10:00:00 发布 · 1.8k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#词云 #python

Python 专栏收录该内容

29 篇文章

订阅专栏

本文介绍了如何使用Python中的WordCloud库，从名为ham.txt的文本文件中提取词频，生成并展示词云图，展示了数据预处理和可视化过程。

说明：基于DataFrame结构的词频表生成词云图

代码

from wordcloud import WordCloud
import matplotlib.pyplot as plt

frequencies = {}
count = 0
for line in open(".\ham.txt"):
    count += 1 
    if count > 1:
        arr = line.split(" ")
        frequencies[arr[0].strip()] = int(arr[1].strip())

wc = WordCloud(
    max_words=100,
    width=400,
    height=400,
)
word_cloud = wc.generate_from_frequencies(frequencies)
# 写词云图片
word_cloud.to_file("wordcloud_ham.jpg")
# 显示词云文件
plt.imshow(word_cloud)
plt.axis("off")
plt.show()