115 lines
3.3 KiB
Python
115 lines
3.3 KiB
Python
#!/usr/bin/env python3.4
|
|
# encoding: utf-8
|
|
"""
|
|
Created on 19-3-20
|
|
@title: '微信好友个性签名词云图展示'
|
|
@author: Xusl
|
|
"""
|
|
import os
|
|
import logging.config
|
|
import matplotlib.pyplot as plt
|
|
import jieba
|
|
import re
|
|
import numpy as np
|
|
|
|
from config import logger_path
|
|
from pro_script.wechat_login import wc_login
|
|
from scipy.misc import imread
|
|
from snownlp import SnowNLP
|
|
from wordcloud import WordCloud, ImageColorGenerator
|
|
|
|
|
|
logging.config.fileConfig(logger_path)
|
|
logger = logging.getLogger("root")
|
|
|
|
pwd_path = os.path.abspath(os.path.dirname(os.getcwd()))
|
|
res_full = os.path.join(pwd_path, 'res/')
|
|
|
|
|
|
def sign_words(fri_infos):
|
|
"""
|
|
好友个性签名
|
|
:param fri_infos:
|
|
:return:
|
|
"""
|
|
func_name = "好友个性签名"
|
|
logger.info('start %s ' % func_name)
|
|
sign_li = []
|
|
rule = re.compile("1f\d+\w*|[<>/=]") # 定义正则规则
|
|
for fri_info in fri_infos:
|
|
signature = fri_info['signature']
|
|
if signature:
|
|
sign_deal = signature.replace('\n', '').replace('\t', '').replace(' ', '')\
|
|
.replace("span", "").replace("class", "").replace("emoji", "")
|
|
sign = rule.sub("", sign_deal)
|
|
sign_li.append(sign)
|
|
|
|
logger.debug('sign_li: %s' % sign_li)
|
|
logger.info('end %s ' % func_name)
|
|
return sign_li
|
|
|
|
|
|
def word_cloud(comment):
|
|
"""
|
|
词云图
|
|
:param comment:
|
|
:return:
|
|
"""
|
|
func_name = "词云图"
|
|
logger.info('start %s ' % func_name)
|
|
conf_path = os.path.join(pwd_path, 'conf/')
|
|
comment_txt = ''
|
|
back_img = plt.imread(conf_path + '/peiqi.jpg')
|
|
cloud = WordCloud(font_path=conf_path + '/simhei.ttf', # 若是有中文的话，这句代码必须添加，不然会出现方框，不出现汉字
|
|
background_color="white", # 背景颜色
|
|
max_words=5000, # 词云显示的最大词数
|
|
mask=back_img, # 设置背景图片
|
|
max_font_size=100, # 字体最大值
|
|
random_state=42,
|
|
width=360, height=591, margin=2, # 设置图片默认的大小,但是如果使用背景图片的话,保存的图片大小将会按照其大小保存,margin为词语边缘距离
|
|
)
|
|
for li in comment:
|
|
comment_txt += ' '.join(jieba.cut(li, cut_all=False))
|
|
wc = cloud.generate(comment_txt)
|
|
image_colors = ImageColorGenerator(back_img)
|
|
plt.figure("wordc")
|
|
plt.imshow(wc.recolor(color_func=image_colors))
|
|
wc.to_file(res_full + '好友个性签名词云图.png')
|
|
logger.info('end %s ' % func_name)
|
|
|
|
|
|
def snowlp_analysis(comment):
|
|
"""
|
|
|
|
:param comment:
|
|
:return:
|
|
"""
|
|
func_name = "snow_analysis"
|
|
logger.info('start %s ' % func_name)
|
|
sentimentslist = []
|
|
for li in comment:
|
|
if len(li) > 0:
|
|
s = SnowNLP(li)
|
|
print(li, s.sentiments)
|
|
sentimentslist.append(s.sentiments)
|
|
fig1 = plt.figure("sentiment")
|
|
plt.hist(sentimentslist, bins=np.arange(0, 1, 0.02))
|
|
plt.savefig(res_full + '好友签名情感分析')
|
|
plt.show()
|
|
logger.info('end %s ' % func_name)
|
|
|
|
|
|
def deal():
|
|
"""
|
|
处理入口
|
|
:return:
|
|
"""
|
|
fri_infos = wc_login()
|
|
sign_li = sign_words(fri_infos)
|
|
word_cloud(sign_li)
|
|
snowlp_analysis(sign_li)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
deal()
|