Python探索微信好友信息

itchat.login()      # 通过二维码连接登录账号
friend_msg = itchat.get_friends(update=True)[0:]        # 获取微信好友信息


# 将微信好友信息保存起来，减少登录的次数
with open('./friend_message.json', 'w', encoding='utf-8') as file:
    json.dump(friend_msg, file, ensure_ascii=False)


file.close()


复制

def get_gender(message):
    sex_dic = {}
    sex_list = []


    for i in range(1, len(message)):
        sex = message[i]['Sex']
        if sex == 1:
            sex_dic['Male'] = sex_dic.get('Male', 0) + 1
            sex_list.append("男")
        elif sex == 2:
            sex_dic['Female'] = sex_dic.get('Female', 0) + 1
            sex_list.append("女")
        else:
            sex_dic['Unknown'] = sex_dic.get('Unknown', 0) + 1
            sex_list.append("Unknown")


    return sex_dic, sex_list
复制

 def get_city(message):
    city_list = []


    for i in range(1, len(message)):
        city = message[i]['City']
        if city == '':
            city_list.append(None)
        else:
            city_list.append(city)


    return city_list
复制

with open("./friend_message.json", 'r', encoding='utf-8') as file:
    f_msg = json.load(file)
file.close()


sex_dic, sex_list = getData.get_gender(f_msg)
city_list = getData.get_city(f_msg)
# 将三个属性组成 DataFrame
data = pd.DataFrame({'gender': sex_list, 'city': city_list})
# 获取城市前十的数量
city_dict = data['city'].value_counts()[:15].to_dict()
# 绘制性别环状图
draw_pie(sex_dic)
# 绘制城市柱状图
draw_bar(city_dict)
复制