Python 标签云拟合图片

笔者最近发现了一个神奇的 Python 库，名为 wordcloud 。借助它，可以轻松地用标签云拼出一副图片，就像上图这样。让我们一起看看具体的步骤吧。

环境

请确保 Python 3 和 pip3（即 python3 -m pip ）都可以正常使用。需要使用 pip 安装的模块有：

• matplotlib
• numpy
• pillow
• requests
• wordcloud

其中 wordcloud 模块有些特殊，笔者尝试执行 pip3 install wordcloud ，结果报错了。如果你也遇到了类似的问题，请执行以下安装方式：

1
2
3
4

git clone https://github.com/amueller/word_cloud.git
cd word_cloud
pip3 install -r requirements.txt
python3 setup.py install

这样就完成安装了。

1
2
3
4
5
6
7
8
9
10

frequencies = {
    "JavaScript": 18,
    "Python": 16,
    "Java": 13,
    "Go": 8,
    "TypeScript": 7,
    "C++": 7,
    "Ruby": 6,
    "PHP": 5
}

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

#!/usr/bin/env python3

import numpy as np
import json
from PIL import Image
from wordcloud import WordCloud, ImageColorGenerator
import matplotlib.pyplot as plt
import requests

r = requests.get("https://cdn.jsdelivr.net/gh/madnight/[email protected]/src/data/gh-star-event.json")
jsondict = json.loads(r.text)

r = requests.get("https://octodex.github.com/images/original.png", stream=True)

pic = np.array(Image.open(r.raw).convert("RGBA"))
pic[np.where(pic[:, :, 3] == 0)] = 255

frequencies = {}
for entry in jsondict:
    if entry["year"] == "2018" and entry["quarter"] == "1":
        frequencies[entry["name"]] = np.log(int(entry["count"]))

image_colors = ImageColorGenerator(pic, [255,255,255])
wc = WordCloud(background_color="white", mask=pic)
wc.generate_from_frequencies(frequencies)

plt.imshow(wc.recolor(color_func=image_colors), interpolation="bilinear")
wc.to_file("octocat.png")
plt.axis("off")
plt.show()