对上次爬取豆瓣的数据进行利用,对评论做成词云图,本次需要用到山慈生成的文件—逐梦.csv 具体看详情点击
还需要一张图片 ti.jpg,图片可以随意一张,有明显的色彩变化即可
需要安装如下包,pip install xxx
import jieba
import csv
from wordcloud import WordCloud
import numpy
from PIL import Image
import jieba
import csv
from wordcloud import WordCloud
import numpy
from PIL import Image
#1.读取文件
def resData():
#词云 简单的数据分析形式,将数据进行筛选和处理
commentList = []
stars = ['很差','较差','还行','推荐','力荐']
# 存放到列表
with open('逐梦.csv','r',encoding='utf-8') as file:
csvReader = csv.reader(file)
for item in csvReader:
#将无用数据 去除掉
if (item[1] in stars):
#将处理好的数据添加到类表中
commentList.append(item[2])
return commentList
#2.生成词云图
def generateWordCloud():
#将列表中的数据 转换成字符串
finalComment = ''
comments = resData()
for comment in comments:
finalComment+=comment
#利用jieba分词库 进行分词 cut lcut
finalComment = ' '.join(jieba.cut(finalComment))
#读取图片 词云时可以自定义图形形状
image = numpy.array(Image.open('ti.jpg'))
#利用wordloud进行设置属性生成词云
word = WordCloud(
font_path='C:/Windows/Fonts/msyhbd.ttc',
background_color='white',
mask=image
).generate(finalComment)
word.to_file('逐梦词云图.jpg')
if __name__ == '__main__':
generateWordCloud()
结果: