爬取微信好友信息

在写程序之前,需要安装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)
  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值