在写程序之前,需要安装Pillow、itchat、pandas三个模块。此三个模块可以直接使用pip进行安装,将生成信息写进csv文件中,或将信息生成字典(有代码),或将每条信息写进自己的数据库都皆可用于分析好友男女比例与获取好友备注、昵称、所在城市以及个性签名等信息。
import itchat
from pandas import DataFrame
itchat.login()
friends = itchat.get_friends()[1:]
# print(friends)
male = female = others = 0
for friend in friends[1:]:
sex = friend['Sex']
if sex == 1:
male += 1
elif sex == 2:
female += 1
else:
others += 1
total = len(friends[1:])
print("男性好友人数:%s 比例:%.2f%%" % (male, float(male/total)*100) + '\n' +
"女性好友人数:%s 比例:%.2f%%" % (female, float(female/total) * 100) + '\n' +
"性别不详好友人数:%s 比例:%.2f%%" % (others, float(others / total) * 100))
# 用于定义各个变量
def get_var(var):
variable = []
for i in friends:
value = i[var]
variable.append(value)
return variable
RemarkName = get_var('RemarkName')
NickName = get_var('NickName')
Sex = get_var('Sex')
Province = get_var('Province')
City = get_var('City')
Signature = get_var('Signature')
data = {'RemarkName': RemarkName, 'NickName': NickName, 'Sex': Sex,
'Province': Province, 'City': City, 'Signature': Signature}
frame = DataFrame(data)
frame.to_csv('data.csv', index=True)
#生成字典
# infos = []
# for var in friends:
# RemarkName = var['RemarkName']
# NickName = var['NickName']
# Sex = var['Sex']
# Province = var['Province']
# City = var['City']
# Signature = var['Signature']
# info = {'RemarkName': RemarkName, 'NickName': NickName, 'Sex': Sex,
# 'Province': Province, 'City': City, 'Signature': Signature }
# infos.append(info)
# print(infos)