利用Python统计微信联系人男女比例以及简单的地区分布

寒暄的话不多说，直接进入主题。

运行效果图：

【准备环境】

　　Python版本：v3.5及其以上

　　开发工具：随意，此处使用Pycharm

【依赖包】

　　1、itchat （CMD运行：pip install itchat   进行安装）

　　2、pycharts （CMD运行：pip install pyecharts   进行安装）

　　itchat包是对网页版微信相关接口封装的一个第三方包，目前来说比较好用，一会代码里面会用到相关接口（注释说明）；

　　pycharts包进行图表的创建，只是用到了其初级功能，大家有时间可以将代码改改，生成更全面直观的图表，代码中使用的柱状图，稍显low，可以升级为全国热点图，这样人员的地区分布就更加直观了。

代码：

import itchat
from collections import Counter
from pyecharts import Bar

dict_sex = {}
count_city = None

# itchat微信登录，hotReload表示热登录，如果是True，下一次就不用扫码了（时间不能过长），会在根目录生成一个 itchat.pkl 的文件
itchat.auto_login(hotReload=True)
# itchat 的get_friends接口，获取微信好友列表，返回的列表第一位是你自己，如果想过滤掉自己，改为：itchat.get_friends()[1:]
member_list = itchat.get_friends()[0:]


def calc_all_sex():
    """
    微信联系人总男女信息
    :return:
    """
    man = woman = others = 0
    city = []
    for index, name in enumerate(member_list):
        print("	{}、{}({})".format(index, name["RemarkName"] if name["RemarkName"] is not "" else name["NickName"], name["UserName"]))
        sex = name["Sex"]
        if sex == 1:
            man += 1
        elif sex == 2:
            woman += 1
        else:
            others += 1
        if name["City"] == "":
            city.append("未知城市")
        else:
            city.append(name["City"])

    global count_city
    count_city = Counter(city)
    total = len(member_list)
    man_percent = (float(man) / total * 100)
    woman_percent = (float(woman) / total * 100)
    others_percent = (float(others) / total * 100)

    print("
>>>>>>>>>>>>>微信联系人总男女信息：")
    print("男性好友：%.2f%%" % man_percent)
    print("女性好友：%.2f%%" % woman_percent)
    print("其    它：%.2f%%" % others_percent)


class PeopleInfo:
    def __init__(self, man_, woman_, _others, total_):
        self.man = man_
        self.woman = woman_
        self.others = _others
        self.total = total_


def count(dict_={}):
    """
    计算各个地区的男女人数
    :param dict_:
    :return:
    """
    print("
>>>>>>>>>>>>>各地区男女分布信息：")
    for val in dict_:
        city_tmp = "" if val == "未知城市" else val
        man = woman = others = 0
        for member in member_list:
            if member["City"] == city_tmp:
                sex = member["Sex"]
                if sex == 1:
                    man += 1
                elif sex == 2:
                    woman += 1
                else:
                    others += 1
        people_info = PeopleInfo(man, woman, others, dict_[val])
        dict_sex[val] = people_info
        print("【{}】男性：{}，女性：{}，其它：{}".format(city_tmp, man, woman, others))


def count_sex_area():
    """
    统计联系人性别、地区
    :return:
    """
    calc_all_sex()
    attr = ["{}".format(i) for i in count_city]
    count(count_city)
    v1 = []
    man_count = []
    woman_count = []
    others_count = []
    for i in attr:
        v1.append(count_city[i])
        man_count.append(dict_sex[i].man)
        woman_count.append(dict_sex[i].woman)
        others_count.append(dict_sex[i].others)

    bar = Bar(title="{}的微信联系人分布".format(member_list[0]["NickName"]), subtitle="微信联系人分布情况", width=2024, height=768)
    bar.add("地区人数", attr, v1, mark_line=["average"], mark_point=["max", "min"])
    bar.add("男性", attr, man_count, mark_line=["average"], mark_point=["max", "min"])
    bar.add("女性", attr, woman_count, mark_line=["average"], mark_point=["max", "min"])
    bar.render(path="地区统计.html")


def get_signatare():
    """
    获取微信联系人的签名信息
    :return:
    """
    for member in member_list:
        signatare = str(member["Signature"])
        print("
{}:
	>>>>>:{}".format(member["RemarkName"], signatare))


if __name__ == '__main__':
    count_sex_area()
    # get_signatare()