词云图怎么做Python

Hey小伙伴们，今天要来分享一个超级有趣的Python小技巧——制作词云图，词云图是一种非常直观的展示文本数据中关键词频率的方式，通过不同的字体大小或颜色来表示词语的重要性，无论是社交媒体分析、文本分析还是任何需要数据可视化的场景，词云图都能派上用场，就让我们一起动手，用Python来制作一个属于自己的词云图吧！

我们需要准备一些基础工具，在Python中，我们可以使用wordcloud库来生成词云图，如果你还没有安装这个库，可以通过pip来安装：

pip install wordcloud

安装完成后，我们就可以开始制作词云图了，我们需要一些文本数据，这里我们以一段简单的文本为例：

text = """
Python 是一种广泛使用的解释型、高级编程、通用型编程语言，它的设计哲学强调代码的可读性，其语言构造及其对象导向的方法旨在帮助程序员编写清晰、逻辑性强的代码。
"""

我们需要使用wordcloud库来生成词云，导入必要的库：

from wordcloud import WordCloud
import matplotlib.pyplot as plt

创建一个WordCloud对象，并传入我们的文本数据：

wordcloud = WordCloud(width=800, height=400, background_color='white').generate(text)

width和height参数用于设置词云图的尺寸，background_color参数用于设置背景颜色。

生成词云图之后，我们可以使用matplotlib库来显示它：

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

这样，我们就得到了一个简单的词云图，我们可以让这个词云图更加个性化和有趣，我们可以指定词云的形状，使用自定义的停用词列表，或者调整词云的颜色。

自定义形状的词云图

如果你想让词云图的形状更加独特，可以传入一个遮罩图像，这样，词云就会根据遮罩的形状来生成，你需要准备一个遮罩图像，比如一个心形的PNG图片，使用以下代码：

from PIL import Image
import numpy as np
加载遮罩图像
mask = np.array(Image.open('mask.png'))
wordcloud = WordCloud(width=800, height=400, background_color='white', mask=mask).generate(text)

使用自定义的停用词列表

我们不希望某些词语出现在词云图中，的”、“是”这样的常用词，这时，我们可以创建一个停用词列表，并在生成词云时传入这个列表：

stopwords = set(['的', '是'])
wordcloud = WordCloud(width=800, height=400, background_color='white', stopwords=stopwords).generate(text)

调整词云的颜色

词云的颜色也可以自定义，我们可以使用color_func参数来指定一个颜色函数，或者使用WordCloud对象的recolor方法来重新着色：

wordcloud.recolor(color_func=lambda *args, **kwargs: (255, 0, 0))  # 将所有词云颜色设置为红色

或者，我们可以使用matplotlib的颜色映射：

from matplotlib import cm
def color_func(word, font_size, position, orientation, random_state=None, **kwargs):
    return cm.Blues(int(font_size))
wordcloud = WordCloud(width=800, height=400, background_color='white', color_func=color_func).generate(text)

这样，我们就可以根据字体大小来着色，使得词云图更加丰富多彩。

记得保存你的词云图：