1将上次获得的评论整理成一个词云,借助了cv2想生成某心形状的词云。
import numpy as np
import pandas as pd
import jieba
import jieba.posseg
import cv2
data_source= pd.DataFrame(pd.read_excel(r'D:\tem\001\comments_lots.xlsx',sheet_name='comments_doubian_真心半解'))
text_half_front = data_source.iloc[0:40,1]#获取第一列中0到40行的元素
text_half_next = data_source.iloc[40:80,1]
comments_half_front = " ".join(text_half_front)#利用join将列表转换为字符串
comments_half_next = " ".join(text_half_next)
word_half_front = jieba.posseg.cut(comments_half_front)#利用jieba.posseg.cut函数进行分词,并得到每个词对应的词性
word_half_next = jieba.posseg.cut(comments_half_next)
word_front = list()#列表容纳词性
word_next = list()
for item in word_half_front:
word_front.append(item)
for item in word_half_next:
word_next.append(item)
word_front_dict = dict(word_front)#将刚存储的词和词性转换为字典
word_next_dict = dict(word_next)
word1_front_lit = list()
word2_next_lit = list()
word_remove = ['ud','ug','uj','ul','uv','uz','r','rr','x',' ']
for key,value in word_front_dict.items():#除去词性为标点符号,的,得等等的词,并再转换为列表
if value not in word_remove:
word1_front_lit.append(key)
for key,value in word_next_dict.items():
if value not in word_remove:
word2_next_lit.append(key)
ciyun_front = " ".join(word1_front_lit)
ciyun_next = " ".join(word2_next_lit)
mask = cv2.imread('D:\Builder_tool\Python\heart1.jpeg')
wc = WordCloud(background_color="white",\
width = 2000,\
height=1400,\
max_words=1000,\
max_font_size=100,\
mask=mask,\
font_path='simfang.ttf',
#设置为中文
).generate(ciyun_next)
wc.to_file('D:\Builder_tool\Python\ciyun_next_digital.jpg')
效果如图:
很显然,并不是心形,可能是因为词太多了,根本放不过来。
另外使用coleection进行词频统计的话,同次数的情况下,英文的优先级似乎更高,也能得到更直观的展示。暂无解决办法。