实例
官网:http://amueller.github.io/word_cloud/
例子
安装库文件
import jieba #分词
from matplotlib import pyplot as plt #绘图
from wordcloud import WordCloud #词云
from PIL import Image #图片处理
import numpy as np #矩阵运算
import sqlite3 #sqlite3数据库
安装问题
由于版本不兼容,需要安装相应版本的wordcloud
第一步:http://www.lfd.uci.edu/~gohlke/pythonlibs/#wordcloud 页面下载所需的wordcloud模块的whl文件,这里选择wordcloud-1.6.0-cp37-cp37m-win32.whl 版本。
注意:下载的版本!!cp后面是你的python版本,
例如:3.7.1就是cp37查看python的版本号,可以在命令提示符界面下输入
pip install 文件名
实例
import jieba #分词
from matplotlib import pyplot as plt #绘图
from wordcloud import WordCloud #词云
from PIL import Image #图片处理
import numpy as np #矩阵运算
import sqlite3 #sqlite3数据库
conn = sqlite3.connect('movie.db')
cur = conn.cursor()
sql = 'select instroduction from movie250'
data = cur.execute(sql)
text = ""
for item in data:
text = item[0] + text
# print(item[0])
print(text)
cur.close()
conn.close()
# 使用jieba分词
cut = jieba.cut(text)
string = ' '.join(cut)
# print(string)
# print(len(string))
#准备图片
img = Image.open(r'蒙娜丽莎.jpg') # 打开遮罩数组
# 矩阵运算,将图片变为数组
img_array = np.array(img)
# 使用wordcloud处理
wc = WordCloud(
background_color='white', #颜色
width=400,
height=800,
mask=img_array, # 数组
font_path='站酷快乐体2016修订版.ttf' # 字体位置C:\Windows\Fonts,字体需要复制到自己的目录下,不然可能找不到
)
# 从切好的里面生成分词
wc.generate_from_text(string)
# 绘制图片
fig = plt.figure(1) # 绘制一张图片
plt.imshow(wc) # 显示生成后的词云
plt.axis('off') # 是否显示坐标图
# plt.show() # 显示生成的慈云图
# 保存到本地
plt.savefig(r'word.jpg',dpi=500)