Python獲取好友地區分佈及好友性別分佈

故事膠片發表於2019-07-10

Python

利用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')
複製程式碼

效果

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")
複製程式碼