词云图

今天我们用jupyter里的wordcloud来生成一个词云图

　　　　　本人觉得很好玩，来看代码。

from wordcloud import WordCloud
from matplotlib import pyplot as plt
from PIL import Image
import numpy as np

#1 准备数据
f = open('豆瓣舌尖中国.txt','rb')
content = f.read()
result = jieba.lcut(content,cut_all=False)
content = ' '.join(result)

#2 初始化词云图对象， pip install wordcloud

background_img = np.array(Image.open('4c192d16b93f86a0c99c19a60aab733.jpg')) #加载图片生成高维数组
wc = WordCloud(background_color='white',
               font_path='C:\Windows\Fonts\SimHei.ttf',
               mask=background_img,  #设置背景
               max_font_size=150,
               max_words=100,  #控制词的个数
               stopwords={"食物"}, #删除词
              )

word = wc.generate(content) #生成词云
plt.imshow(word)
plt.axis('off') #关闭坐标
plt.show()

效果如下图