Python word_cloud生成词云

需要使用到的各类库

• PIL
• matplotlib
• numpy
• wordcloud

首先是需要安装wordcloud
pip install wordcloud
即可安装最新的wordcloud库

开始构成词云

• 读取需要用来构成词云的文字

#使用了《爱丽丝梦游仙境》作为词的来源
with open("alice.txt") as f:
    text=f.readlines()
    text_clean=[]
    for line in text:
        line=line.strip()
        text_clean.append(line.replace("\n",""))
text="".join(text_clean)

• 读取需要用来做mask的图片（可选）

import numpy as np
import PIL.Image as Image
coloring=np.array(Image.open("path"))

• 构建word_cloud对象的实例

from wordcloud import WordCloud,ImageColorGenerator
my_wordcloud=WordCloud(backgroud_color=" ",max_words=任意数字，
mask=上一步的coloring,max_font_size=自己设定,random_state=变化的范围,scale=生成图片和原图的大小倍数,font_path=字体路径)
#example
my_wordcloud = WordCloud(background_color="white", max_words=20000,mask=coloring,
                         max_font_size=40, random_state=42, scale=2,
                         font_path=" Fonts/Hannotate TC/Hannotate.ttc")

• 构成词云

#构成词云
my_wordcloud.generator(text)

• 绘制出词云

import matplotlib.pyplot as plt
plt.imshow(my_wordcloud)
plt.axis("off")
plt.show()

默认颜色的词云

默认颜色

#使用图片背景颜色构成词云

image_colors=ImageColorGenerator(coloring)
plt.imshow(my_wordcloud.recolor(color_func=image_colors))
#plt.imshow(my_wordcloud)
plt.axis("off")
plt.show()