轻松抓住内容关键字！

永恒君

发布于 2022-12-07 15:18:05

2090

发布于 2022-12-07 15:18:05

文章被收录于专栏：开源小分队开源小分队

大家好，我是开源大叔！今天给大家分享一个非常好用的行为词云图开源项目：wordcloud。

介绍

当我们看到一篇文章或者电子书的时候，想知道文章或书中主要内容，则可以采用词云图来处理。词云图就是对文本中出现频率较高的关键词进行可视化的展现。接下来跟着大叔来看看怎么使用吧。

快速入门

在使用之前我们先安装一下 wordcloud，这里安装使用的是 pip3，当然也可以直接使用 pip 进行安装。

pip3 install wordcloud

我们先来一个简单的例子，如下：

import numpy as np
import matplotlib.pyplot as plt
from wordcloud import WordCloud

text = "Hello wordcloud"

x, y = np.ogrid[:300, :300]

mask = (x - 150) ** 2 + (y - 150) ** 2 > 130 ** 2
mask = 255 * mask.astype(int)


wc = WordCloud(background_color="white", repeat=True, mask=mask)
wc.generate(text)

plt.axis("off")
plt.imshow(wc, interpolation="bilinear")
plt.show()

运行效果图如下：

上面展示了英文的词语，我们去试一下中文看看效果如何。

  d = path.dirname(__file__) if "__file__" in locals() else os.getcwd()
  text = open(path.join(d, 'info.txt')).read()
  alice_mask = np.array(Image.open(path.join(d, "file/alice_mask.png")))
  stopwords = set(STOPWORDS)
  stopwords.add("said")
  wc = WordCloud(font_path="Hiragino Sans GB.ttc",background_color="white", max_words=2000, mask=alice_mask,
                 stopwords=stopwords, contour_width=3, contour_color='steelblue')
  wc.generate(text)
  wc.to_file(path.join(d, "alice.png"))
  plt.imshow(wc, interpolation='bilinear')
  plt.axis("off")
  plt.figure()
  plt.imshow(alice_mask, cmap=plt.cm.gray, interpolation='bilinear')
  plt.axis("off")
  plt.show()

运行效果如下：