python词云 wordcloud 简单实例 入门

1. 引言

词云（Word Cloud）是一种直观展示文本数据的可视化方法，通过对文本中出现频率较高的词语进行可视化展示，帮助人们快速了解文本的关键内容。Python中有很多库可以用来生成词云，其中最流行的就是wordcloud库。

2. 安装wordcloud库

在使用wordcloud库之前，我们需要先安装它。可以通过以下命令来安装wordcloud：

pip install wordcloud

3. 简单实例

3.1 创建词云对象

首先，我们需要导入wordcloud库和matplotlib库：

import wordcloud

import matplotlib.pyplot as plt

然后，我们可以创建一个词云对象：

cloud = wordcloud.WordCloud()

3.2 生成词云

接下来，我们需要准备一段文本数据，并将其传入词云对象中进行处理：

text = "Python是一种高级编程语言，简单而强大。"

cloud.generate(text)

生成的词云对象中已经包含了词云的图片数据。

3.3 显示词云

最后，我们可以使用matplotlib库中的imshow()方法来显示词云：

plt.imshow(cloud, interpolation='bilinear')

plt.axis('off')
plt.show()

4. 高级用法

4.1 设置词云样式

在创建词云对象时，可以传入一些参数来设置词云的样式，如背景颜色、字体、最大词数等。

cloud = wordcloud.WordCloud(background_color='white', max_words=100, font_path='msyh.ttf')

在上面的例子中，我们设置了词云的背景颜色为白色，最大词数为100，并使用了微软雅黑字体。

4.2 设置词云形状

除了默认的矩形形状，词云还可以根据自定义的形状来生成。首先，我们需要准备一个自定义形状的图片文件：

from PIL import Image

import numpy as np
mask = np.array(Image.open("cloud_shape.png"))

然后，我们可以在创建词云对象时传入该数组作为参数：

cloud = wordcloud.WordCloud(mask=mask)

生成的词云会按照图片中的形状进行显示。

4.3 设置词云颜色

我们还可以使用collocations属性来设置词云的颜色。该属性的取值可以是一个颜色名称或一个颜色函数。

cloud = wordcloud.WordCloud(collocations=False, colormap='Paired')

在上面的例子中，我们禁用了词云中的重复词组，并使用了Paired颜色映射。

5. 总结

通过本文的介绍，我们学习了如何使用python的wordcloud库来生成词云图。我们了解了基本的使用方法，并简单介绍了一些高级用法。希望本文可以帮助读者快速入门词云的生成，并在实际项目中发挥作用。