天天看點

itchat擷取微信好友資訊

itchat擷取微信好友資訊

itchat功能很強大,網上有關itchat的資料很多,使用起來也很有趣。介于技術修養還不夠深厚,本文隻是介紹了冰山一角,更多更有趣的功能日後還需要多多學習。

本文介紹了使用itchat接口擷取微信好友清單中男性和女性朋友的比例,以及他們的基本資訊,包括地區、性别、省份、簽名等等。最後把這些好友資訊寫入到csv檔案中,能夠很清晰直白地做一些分析(比如自己男性朋友居多,還是女性朋友居多)

參照一些資料,自己也做了一點點修改,demo如下:

#!/usr/bin/env python 
# -*- coding:utf-8 -*-

import itchat
from pandas import DataFrame

itchat.auto_login(True)
friends = itchat.get_friends(update=True)[0:]
# print(friends)
total = len(friends[1:])#從1開始,0是自己不能算
#擷取男女性朋友比例
def male_female_ratio():
    male = female = other = 0  # 初始化計數器
    for i in friends:
        sex = i['Sex']
        if sex == 1:
            male += 1
        elif sex == 2:
            female += 1
        else:
            other += 1
    # print('男性比例:%.2f%%' + '女性比例:%.2f%%' + '其他比例:%.2f%%' %(male/total*100,female/total*100,other/total*100))
    print('男性比例:%.2f%%' %(male/total*100) + "\n" + '女性比例:%.2f%%' %(female/total*100) + "\n" + '其他比例:%.2f%%' %(other/total*100))

def get_data(arg):
    data_list = []
    for i in friends:
        value = i[arg]
        if value == 1:#1代表男/2代表女
            value = '男'#這裡把1/2換成了男女
            data_list.append(value)
        elif value == 2:
            value = '女'
            data_list.append(value)
        elif value == 0:
            value = '未知'
            data_list.append(value)
        else:
            data_list.append(value)
    return data_list

def ger_friend_msg():
    import time
    RemarkName = get_data('RemarkName')
    NickName = get_data('NickName')  # 是一個清單
    City = get_data('City')
    Province = get_data('Province')
    Sex = get_data('Sex')
    Signature = get_data('Signature')
    data = {'RemarkName(備注)': RemarkName, 'Nickname(昵稱)': NickName, 'City(城市)': City, 'Province(省份)': Province,
            'Sex(性别)': Sex, 'Signature(微信簽名)': Signature}  # 字典,字典的value是清單
    frame = DataFrame(data)
    time = time.strftime("%Y-%m-%d_%H_%M_%S", time.localtime())  # 生成一個目前事件,以這個時間來命名最後地csv檔案,能很直白地看出最新執行腳本的時間和最新的csv檔案
    file_name = time + 'data.csv'
    frame.to_csv(file_name, encoding='utf_8_sig', index=True)

if __name__ == '__main__':
    male_female_ratio()
    ger_friend_msg()
           

運作環境win10+python3

最終的結果如下:

列印出的比例:

itchat擷取微信好友資訊

csv截圖:

itchat擷取微信好友資訊

這個腳本還有可以優化的地方,後期空了再做些小優化吧。

一步一步總會成功的。

繼續閱讀