利用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("\t{}、{}({})".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("\n>>>>>>>>>>>>>微信聯系人總男女信息：")
    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("\n>>>>>>>>>>>>>各地區男女分布信息：")
    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("\n{}:\n\t>>>>>:{}".format(member["RemarkName"], signatare))


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