用Python绘制了若干张词云图，惊艳了所有人("Python制作精美词云图，惊艳全场！")

原创

ithorizon 7个月前 (10-20) 阅读数 23 #后端开发

Python制作精美词云图，惊艳全场！

一、引言

在当今这个信息爆炸的时代，数据的可视化变得越来越重要。词云图作为一种简洁直观的数据可视化工具，可以将大量文本数据中的关键词以云状图形展示出来，使文本数据的特征和关键词一目了然。本文将介绍怎样使用Python来制作精美的词云图，并分享一些实用的技巧和心得。

二、词云图简介

词云图，也称为文字云，是一种将文本数据中的关键词按照频率和重要性以不同大小和颜色的文字展示出来的图形。在词云图中，字体大小和颜色可以反映关键词的重要性和出现频率，使人们可以敏捷了解文本的主题和重点。

三、Python制作词云图的基础步骤

下面，我们将介绍使用Python制作词云图的基础步骤。

1. 准备环境

首先，确保你的Python环境中安装了以下库：

wordcloud：用于生成词云图的核心库。

matplotlib：用于显示和保存词云图。

jieba：用于中文文本的分词。

2. 导入库


import wordcloud
import matplotlib.pyplot as plt
import jieba

3. 准备文本数据

假设我们有一段中文文本数据，需要先进行分词处理。


text = "Python是一种广泛使用的高级编程语言，其设计哲学强调代码的可读性和简洁的语法。"
words = jieba.cut(text)

4. 生成词云图

使用wordcloud库生成词云图，并使用matplotlib显示。


wordcloud_text = ' '.join(words)
wordcloud_image = wordcloud.WordCloud(font_path='simhei.ttf',  # 设置字体
                                      width=800, height=600,  # 设置画布大小
                                      background_color='white').generate(wordcloud_text)
plt.imshow(wordcloud_image, interpolation='bilinear')
plt.axis('off')  # 关闭坐标轴
plt.show()

四、进阶技巧

除了基础的词云图，我们还可以通过以下技巧来制作更加精美的词云图。

1. 自定义颜色和形状

wordcloud库允许我们自定义词云图的颜色和形状。


mask = plt.imread('heart.png')  # 读取形状图片
wordcloud_image = wordcloud.WordCloud(font_path='simhei.ttf',
                                      width=800, height=600,
                                      background_color='white',
                                      mask=mask).generate(wordcloud_text)
plt.imshow(wordcloud_image, interpolation='bilinear')
plt.axis('off')
plt.show()

2. 使用自定义词频

我们可以基于自定义的词频来生成词云图，使某些关键词更加突出。


word_freq = {'Python': 10, '编程': 5, '简洁': 3, '高级': 2}
wordcloud_text = ' '.join(words)
wordcloud_image = wordcloud.WordCloud(font_path='simhei.ttf',
                                      width=800, height=600,
                                      background_color='white',
                                      relative_scaling=0.5).generate_from_frequencies(word_freq)
plt.imshow(wordcloud_image, interpolation='bilinear')
plt.axis('off')
plt.show()

3. 添加动画效果

虽然wordcloud库本身不赞成动画效果，但我们可以使用matplotlib的动画功能来添加动画效果。


from matplotlib.animation import FuncAnimation
def update(frame):
    wordcloud_image.set_array(wordcloud.WordCloud(font_path='simhei.ttf',
                                                 width=800, height=600,
                                                 background_color='white',
                                                 relative_scaling=0.5).generate_from_frequencies(word_freq))
fig, ax = plt.subplots()
wordcloud_image = ax.imshow(wordcloud.WordCloud(font_path='simhei.ttf',
                                               width=800, height=600,
                                               background_color='white',
                                               relative_scaling=0.5).generate_from_frequencies(word_freq),
                            interpolation='bilinear')
ax.axis('off')
ani = FuncAnimation(fig, update, frames=range(100), blit=True)
plt.show()