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
最終的結果如下:
列印出的比例:
csv截圖:
這個腳本還有可以優化的地方,後期空了再做些小優化吧。
一步一步總會成功的。