Python 获取微信好友地区、性别、签名信息并将结果可视化

欢迎关注天善智能，我们是专注于商业智能BI，人工智能AI，大数据分析与挖掘领域的垂直社区，学习，问答、求职一站式搞定！

对商业智能BI、大数据分析挖掘、机器学习，python，R等数据领域感兴趣的同学加微信：tstoutiao，邀请你进入数据爱好者交流群，数据爱好者们都在这儿。

作者：肖涛  Python爱好者社区专栏作者  个人公众号：GankSharer

个人网站: https://inspurer.github.io/

简书: https://www.jianshu.com/u/1b872cf08f32   

序言

我之前在CSDN博客上分享过两篇有关itchat库的教程:

itchat库实用教程 (一): Python从编程零基础打造一款微信聊天自动回复机器人

https://blog.csdn.net/ygdxt/article/details/79766197

itchat库实用教程 (二): Python使用itchat库+图灵机器人实现群聊助手

https://blog.csdn.net/ygdxt/article/details/81101020

本篇博客是itchat库系列的第三篇文章，它主要实现的功能是获取微信好友地区、性别、签名信息并将结果可视化

正文

根据itchat库获得好友信息

使用下面两行代码可以做到：

itchat.auto_login(hotReload=True)
friends = itchat.get_friends(update=True)

第一行代码是使用itchat库登录微信网页版，它会弹出扫码登录界面,设置参数hotReload=True的目的是短时间内退出程序，再次登录可以不用扫码。

get_friends()返回的是一个list，里面有n个字典项，其中n等于你的微信好友数，由于list类型是可序列化的，我们可以使用json库把它保存到文件中

# 把json对象转成字符串并保存在本地
with open("data.json", "w", encoding="utf-8") as f:
    f.write(json.dumps(friends, indent=2, ensure_ascii=False))

其结果如下:

提取数据并可视化

性别信息可视化

使用下面的代码可以获得所有好友的性别信息

 for friend in friends:
            if friend["Sex"] == 0:
                unknow += 1
            elif friend["Sex"] == 1:
                male += 1
            else:
                female +=1

可视化的结果：

男同学占了大部分，女同学才占1/4，我果然是一个没有故事的男同学。

性别未知是没有设置性别的好友。

地区信息可视化

使用下面的代码可以获得所有好友的地区信息

dic_province = dict()
        for friend in self.friends:
            key = friend["Province"]
            if key not in dic_province.keys():
                dic_province[key] = 1
            else:
                dic_province[key] += 1
        key_list = []
        value_list = []
        for key,value in dic_province.items():
            if key == "":
                key = "其他地区"
            key_list.append(key)
            value_list.append(value)

可视化的结果是:

呃，好像暴露了我自己的地区。

城市信息可视化

获取城市信息和地区类似，

    dic_city = dict()
    for friend in self.friends:
        key = friend["City"]
        if key not in dic_city.keys():
            dic_city[key] = 1
        else:
            dic_city[key] += 1

    data = []
    max_value = 0
    for key,value in dic_city.items():
        # 其他地区
        if len(key) == 0:
            continue
        # 过滤英文等非市级
        if len(key) >= 3:
            continue
        data.append(tuple((key,value)))
        if value > max_value:
            max_value = value

可视化效果如下：

长沙和衡阳的好友最多，信息格式为:城市-经度-纬度-人数

签名信息可视化

使用下面的代码获得所有好友的个性签名

all_sign = ""
        for friend in self.friends:
            sign = friend["Signature"]
            if len(sign)>0:
                # 过滤表情，否则会对词云造成影响
                emoji = re.findall("<span class=.*></span>",sign,re.S)
                if len(emoji) > 0:
                    sign = sign.replace(emoji[0],"")
                all_sign = all_sign + sign + "\n"
        if os.path.exists("wc/sign.txt"):
            os.remove("wc/sign.txt")
        with open("wc/sign.txt","a+",encoding="utf-8") as f:
            f.write(all_sign)

使用词云可视化的结果是

哈哈，也是一群积极向上，积极探讨人生的小伙伴们呢

其他数据可视化

道理类似，不赘述了。

运行示意

运行项目根目录下的main.py文件，可以得到性别和地区可视化结果，如下图：

如果近期没有登录过，程序还会让你扫码登录的

如果需要可视化好友个性签名，则还需要运行项目根目录下wc文件夹下的wc.py文件，如下图：

这个词云是可以定制的，只要替换项目根目录下wc/image下的图片bc.png，就能修改词云的背景，注意最好是纯色背景的图。

所有源代码

微信公众号回复关键词"微信好友"即可获得。

Python的爱好者社区历史文章大合集：

2018年Python爱好者社区历史文章合集（作者篇）

2018年Python爱好者社区历史文章合集（类型篇）

关注后在公众号内回复“ 课程 ”即可获取：

小编的转行入职数据科学（数据分析挖掘/机器学习方向）【最新免费】

小编的Python的入门免费视频课程！

小编的Python的快速上手matplotlib可视化库！

崔老师爬虫实战案例免费学习视频。

陈老师数据分析报告扩展制作免费学习视频。

玩转大数据分析！Spark2.X + Python精华实战课程免费学习视频。