Python获取好友地区分布及好友性别分布情况代码详解_Python

Python获取好友地区分布及好友性别分布情况代码详解

2021-08-07 00:39故事胶片 Python

利用Python + wxpy 可以快速的查询自己好友的地区分布情况，以及好友的性别分布数量。还可以批量下载好友的头像，拼接成大图。感兴趣的朋友跟随小编一起看看吧

利用Python + wxpy 可以快速的查询自己好友的地区分布情况，以及好友的性别分布数量。还可以批量下载好友的头像，拼接成大图。

本次教程是基于上次机器人后的，所有依赖模块都可以复用上次的，还不知道的小伙伴可以戳这里。

python + wxpy 机器人

准备工作

编辑器
一个注册一年以上的微信号

公共部分代码

				?

									from wxpy import * // wxpy 依赖

									from PIL import Image  // 二维码登录依赖

									import os  // 本地下载依赖

									import math

									import webbrowser

									from pyecharts import Map  // 地图展示依赖

									from pyecharts import Pie  // 饼状图依赖

1. 批量下载好友头像

				?

									# 创建头像存放文件夹

									def avaterPath():

									  avaterDir = os.path.join(os.getcwd(), 'wechat')

									  if not os.path.exists(avaterDir):

									    os.mkdir(avaterDir)

									  return avaterDir

									# 获取所有的好友头像并保存

									def saveWxAvater(avaterDir):

									  bot = Bot(cache_path=True)

									  allFriends = bot.friends(update=True)

									  num = 0

									  for friend in allFriends:

									    friend.getAvatar(os.path.join(avaterDir,f'{str(num)}.jpg'))

									    print("好友昵称：%s"%friend.name)

									    num += 1

									# 拼接头像

									def joinAvatar(path):

									  # 获取文件夹内头像个数

									  length = len(os.listdir(path))

									  # 设置画布大小

									  image_size = 2560

									  # 设置每个头像大小

									  each_size = math.ceil(2560 / math.floor(math.sqrt(length)))

									  # 计算所需各行列的头像数量

									  x_lines = math.ceil(math.sqrt(length))

									  y_lines = math.ceil(math.sqrt(length))

									  image = Image.new('RGB', (each_size * x_lines, each_size * y_lines))

									  x = 0

									  y = 0

									  for (root, dirs, files) in os.walk(path):

									    for pic_name in files:

									      try:

									        with Image.open(os.path.join(path, pic_name)) as img:

									          img = img.resize((each_size, each_size))

									          image.paste(img, (x * each_size, y * each_size))

									          x += 1

									          if x == x_lines:

									            x = 0

									            y += 1

									      except Exception as e:

									        print(F"头像读取失败，错误：{e}")

									  img = image.save(os.path.join(os.getcwd(), 'wechat.png'))

									  print('wx好友头像拼接完成!')

									if __name__ == '__main__':

									  avatarDir = avaterPath()

									  saveWxAvater(avatarDir)

									  joinAvatar(avatarDir)

2. 获取好友性别分布

				?

									bot = Bot(cache_path=True) # 弹出二维码登录微信，生成bot对象

									allFriends = bot.friends() # 获取所有的微信好友信息

									type = ['男同学','女同学','外星人'] # 男/女/未知性别好友名称

									v = [0, 0, 0] # 初始化对象好友数量

									# 遍历所有好友，判断该好友性别

									for friend in friends:

									  if friend.sex == 1:

									    v[0] += 1

									  elif friend.sex == 2:

									    v[1] += 1

									  else:

									    v[2] += 1

									pie = Pie("好友性别分布")

									pie.add("", type, v, is_label_show=True)

									pie.render("sex.html")

									webbrowser.open('sex.html')

效果

Python获取好友地区分布及好友性别分布情况代码详解

3. 获取好友地区分布情况

代码部分：

				?

									bot = Bot(cache_path=True) # 弹出二维码登录微信，生成bot对象

									allFriends = bot.friends() # 获取所有的微信好友信息

									areaDic = {} # 定义一个空字典，用于存放省市以及省市人数

									for friend in allFriends:

									  if friend.province not in areaDic:

									    areaDic[friend.province] = 1

									  else:

									    areaDic[friend.province] += 1

									keys = area_dic.keys()

									v = area_dic.values()

									map = Map("好友地域分布", width=1200, height=600)

									map.add("好友地域分布" ,keys, v, maptype='china', is_visualmap=True)

									map.render("area.html")

									webbrowser.open("area.html")