词云图
今天我们用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()
效果如下图