python生成词云

写代码前准备：

需要下载的插件：pandas、pillow、wordcloud、matplotlib、jieba

代码

import jieba
import wordcloud
from matplotlib import pyplot as plt
from PIL import Image
import numpy as np
import pandas as pd

#词云生成



##添加的代码,把刚刚你保存好的图片用Image方法打开,
##然后用numpy转换了一下
mask = Image.open("foodName.jpg")
masks=np.array(mask)

# 读取excel
name=pd.read_csv('kfc.csv')
name_list=name['name'].tolist()
# jieba分词
text=''.join(name_list)
txt=' '.join(jieba.cut(text))


# 设置词云参数
w = wordcloud.WordCloud(background_color = 'white',width = 1000,height = 700,font_path = 'SIMLI.TTF',mask = masks).generate(txt)

#生成图片
plt.imshow(w)
w.to_file('wordcloud.png')