137 lines
5.2 KiB
Python
137 lines
5.2 KiB
Python
#!/usr/bin/env python3.4
|
||
# encoding: utf-8
|
||
"""
|
||
Created on 19-3-25
|
||
@title: '微信群好友信息分析'
|
||
@author: Xusl
|
||
"""
|
||
import os
|
||
import itchat
|
||
import pandas as pd
|
||
import logging.config
|
||
import matplotlib.pyplot as plt
|
||
|
||
from config import logger_path
|
||
from pro_script.wechat_login import wc_login
|
||
|
||
logging.config.fileConfig(logger_path)
|
||
logger = logging.getLogger("root")
|
||
|
||
|
||
def grp_sex_ratio(grp_nm, ):
|
||
"""
|
||
群好友性别比例
|
||
:param grp_nm: 想获取群聊信息的群名
|
||
:return:
|
||
"""
|
||
func_name = "群好友性别"
|
||
logger.info('start %s ' % func_name)
|
||
df_member = pd.read_csv("df_member.csv")
|
||
mem_sex = dict(df_member['Sex'].replace({1: '男', 2: '女', 0: '其他'}).value_counts(normalize=True)) # 使用pandas库自带的统计值函数
|
||
sex_li = []
|
||
proportion = []
|
||
for key, value in mem_sex.items():
|
||
sex_li.append(key)
|
||
proportion.append(format(value, '.2'))
|
||
|
||
plt.rcParams['font.sans-serif'] = ['SimHei'] # 用来正常显示中文标签
|
||
plt.rcParams['axes.unicode_minus'] = False # 用来正常显示负号
|
||
plt.figure(figsize=(5, 5)) # 绘制的图片为正圆
|
||
radius = [0.01, 0.01, 0.01] # 设定各项距离圆心n个半径
|
||
colors = ['red', 'yellowgreen', 'lightskyblue']
|
||
logger.debug('proportion:%s' % proportion)
|
||
|
||
plt.pie(proportion, explode=radius, labels=sex_li, colors=colors, autopct='%.2f%%') # 绘制饼图
|
||
# 加入图例 loc = 'upper right' 位于右上角 bbox_to_anchor=[0.5, 0.5] # 外边距 上边 右边 borderaxespad = 0.3图例的内边距
|
||
plt.legend(loc="upper right", fontsize=10, bbox_to_anchor=(1.1, 1.1), borderaxespad=0.3)
|
||
|
||
plt.title(grp_nm + '群好友性别比例') # 绘制标题
|
||
# 获取上一层目录
|
||
pwd_path = os.path.abspath(os.path.dirname(os.getcwd()))
|
||
desc_full = os.path.join(pwd_path, 'res')
|
||
plt.savefig(desc_full + '/' + grp_nm + '好友性别比例') # 保存图片
|
||
plt.show()
|
||
logger.info('end %s ' % func_name)
|
||
|
||
|
||
def grp_city(grp_nm):
|
||
"""
|
||
群好友城市比例
|
||
:param grp_nm: 想获取群聊信息的群名
|
||
:return:
|
||
"""
|
||
func_name = "群好友城市比例"
|
||
logger.info('start %s ' % func_name)
|
||
df_member = pd.read_csv("df_member.csv")
|
||
df_member = df_member['DisplayName'].values
|
||
mem_nm_li = []
|
||
for mem_nm in df_member:
|
||
if mem_nm:
|
||
mem_nm = str(mem_nm).replace(' ', '')
|
||
mem_nm = mem_nm.replace('一', '-').replace('_', '-').replace('－', '-').replace(' ', ' ')\
|
||
.replace(' ', '-').replace('~', '-').replace('—', '-').replace('+', '-').replace('～','-')
|
||
mem_nm = mem_nm.replace('--', '-')
|
||
mem_nm_li.append(mem_nm)
|
||
logger.debug('mem_nm_li：%s' % mem_nm_li)
|
||
|
||
city_dict, industry_dict = {}, {}
|
||
for per_info in mem_nm_li:
|
||
split_num = per_info.count("-", 0, len(per_info))
|
||
if split_num == 2:
|
||
nm, city, industry = per_info.split('-')
|
||
if city and city not in city_dict.keys():
|
||
city_dict[city] = 1
|
||
elif city:
|
||
city_dict[city] += 1
|
||
|
||
logger.debug('city_dict：%s' % city_dict)
|
||
city_dict_top20 = sorted(city_dict.items(), key=lambda x: x[1], reverse=True)[0:20]
|
||
|
||
city_nm, city_num = [], [] # 城市名 + 数量
|
||
for prov_data in city_dict_top20:
|
||
city_nm.append(prov_data[0])
|
||
city_num.append(prov_data[1])
|
||
pwd_path = os.path.abspath(os.path.dirname(os.getcwd()))
|
||
desc_full = os.path.join(pwd_path, 'res')
|
||
colors = ['#00FFFF', '#7FFFD4', '#F08080', '#90EE90', '#AFEEEE',
|
||
'#98FB98', '#B0E0E6', '#00FF7F', '#FFFF00', '#9ACD32',
|
||
'#00FFFF', '#7FFFD4', '#F08080', '#90EE90', '#AFEEEE',
|
||
'#98FB98', '#B0E0E6', '#00FF7F', '#FFFF00', '#9ACD32'
|
||
]
|
||
plt.rcParams['font.sans-serif'] = ['SimHei'] # 用来正常显示中文标签
|
||
plt.rcParams['axes.unicode_minus'] = False # 用来正常显示负号
|
||
|
||
index = range(len(city_num))
|
||
plt.bar(index, city_num, color=colors, width=0.5, align='center')
|
||
|
||
plt.xticks(range(len(city_nm)), city_nm, fontsize=8) # 横坐轴标签
|
||
for x, y in enumerate(city_num):
|
||
# 在柱子上方1.0处标注值
|
||
plt.text(x, y + 1.0, '%s' % y, ha='center', fontsize=10)
|
||
plt.ylabel('群好友人数') # 设置纵坐标标签
|
||
prov_title = '中产之路2群(新)群好友人数Top20'
|
||
plt.title(prov_title) # 设置标题
|
||
plt.savefig(desc_full + '/中产之路2群(新)群好友人数Top20') # 保存图片
|
||
logger.info('end %s ' % func_name)
|
||
|
||
|
||
def deal():
|
||
"""
|
||
处理入口
|
||
:return:
|
||
"""
|
||
wc_login()
|
||
# grp_lost = itchat.get_chatrooms() # 获取所有群聊信息
|
||
grp_nm = '中产之路2群(新)'
|
||
mid_road = itchat.search_chatrooms(name=grp_nm) # 获取群聊名为中产之路2群(新)的群好友信息
|
||
# 群聊用户列表的获取方法为update_chatroom, detailedMember=True将返回指定用户的信息组成的列表
|
||
member_list = itchat.update_chatroom(mid_road[0]['UserName'], detailedMember=True)
|
||
df_member = pd.DataFrame(member_list['MemberList']) # 将用户信息转化为dateFrame格式
|
||
df_member.to_csv('df_member.csv')
|
||
grp_sex_ratio(grp_nm)
|
||
grp_city(grp_nm)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
deal()
|