80 lines
2.2 KiB
Python
80 lines
2.2 KiB
Python
|
#!/usr/bin/env python3.5
|
||
|
|
# encoding: utf-8
|
||
|
|
"""
|
||
|
|
Created on 19-3-18
|
||
|
|
@title: '探索好友性别比例饼图'
|
||
|
|
@author: Xusl
|
||
|
|
"""
|
||
|
|
import os
|
||
|
|
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 sex_ratio(fri_infos):
|
||
|
|
"""
|
||
|
|
好友性别比例
|
||
|
|
:param fri_infos:
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
func_name = "好友性别比例"
|
||
|
|
logger.info('start %s ' % func_name)
|
||
|
|
total = len(fri_infos) # 好友总数量
|
||
|
|
man = 0
|
||
|
|
woman = 0
|
||
|
|
other = 0
|
||
|
|
|
||
|
|
for fri_info in fri_infos:
|
||
|
|
sex = fri_info['sex']
|
||
|
|
# 如果sex=1 代表男性 sex=2代表女性
|
||
|
|
if sex == 1:
|
||
|
|
man += 1
|
||
|
|
elif sex == 2:
|
||
|
|
woman += 1
|
||
|
|
else:
|
||
|
|
other += 1
|
||
|
|
|
||
|
|
man_ratio = int(man)/total * 100
|
||
|
|
woman_ratio = int(woman)/total * 100
|
||
|
|
other_ratio = int(other)/total * 100
|
||
|
|
plt.rcParams['font.sans-serif'] = ['SimHei'] # 用来正常显示中文标签
|
||
|
|
plt.rcParams['axes.unicode_minus'] = False # 用来正常显示负号
|
||
|
|
plt.figure(figsize=(5, 5)) # 绘制的图片为正圆
|
||
|
|
sex_li = ['男', '女', '其他']
|
||
|
|
radius = [0.01, 0.01, 0.01] # 设定各项距离圆心n个半径
|
||
|
|
colors = ['red', 'yellowgreen', 'lightskyblue']
|
||
|
|
proportion = [man_ratio, woman_ratio, other_ratio]
|
||
|
|
|
||
|
|
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('微信好友性别比例') # 绘制标题
|
||
|
|
# 获取上一层目录
|
||
|
|
pwd_path = os.path.abspath(os.path.dirname(os.getcwd()))
|
||
|
|
desc_full = os.path.join(pwd_path, 'res')
|
||
|
|
plt.savefig(desc_full + '/微信好友性别比例') # 保存图片
|
||
|
|
plt.show()
|
||
|
|
logger.info('end %s ' % func_name)
|
||
|
|
|
||
|
|
|
||
|
|
def deal():
|
||
|
|
"""
|
||
|
|
处理入口
|
||
|
|
:return:
|
||
|
|
"""
|
||
|
|
fri_infos = wc_login()
|
||
|
|
sex_ratio(fri_infos)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
deal()
|
||
|
|
|