微信好友分布分析

Posted 2021-12-18 justlikecode

一.环境要求

Windows 10；python 3.7；

第三方库

wxpy 0.3.9.8——登录微信
openpyxl 2.6.2——可对 excel 表格进行操作

pyecharts 0.5.11——百度的开源可视化库，

wordcloud 1.5.0——词云制作库

matplotlib 3.1.0——生成词云图

pandas 0.24.2——读取 excel 表格的数据

pyecharts-snapshot 0.2.0——是 pyecharts 的依赖库之一

echarts-countries-pypkg——pyecharts 的世界地图包

echarts-china-provinces-pypkg——pyecharts 的中国省份地图包

 

第三方库的安装

（2019年6月10号）除了 pyecharts 其它都可以直接使用 pip 安装，而安装 pycharts 需要特殊些(因为最新版已出为 1.X版，不向下兼容)，如下：

pip install wxpy
pip install openpyxl
pip install PIL
pip install pandas
pip install wordcloud

pip install pyecharts==0.5.11
pip install echarts-countries-pypkg
pip install echarts-china-provinces-pypkg
pip install pyecharts-snapshot

　　

二.分析思路

 1.登录微信，获取好友的基本信息

使用 wxpy 的 Bot() 模块，登录微信

 

from wxpy import *

bot = Bot(cache_path = True)

#获取全部好友的信息
friend_all = bot.friends()

#输出 friend_all 看一下
print(friend_all)

#输出好友数量
#print(len(friend_all))

 

输出结果为

 

 

2.把好友信息存为更易处理的二维列表

 

#建立一个二维列表，存储基本好友信息
lis = [[‘NickName‘,‘Sex‘,‘City‘,‘Province‘,‘Signature‘,‘HeadImgUrl‘,‘HeadImgFlag‘]]
#把好有特征数据保存为列表
for a_friend in friend_all:
    #遍历 friend_all, 并使用 raw.get(‘ ‘, None), 来获取每一各朋友的基本信息
    NickName = a_friend.raw.get(‘NickName‘, None)
    Sex = 1:"男", 2:"女", 0:"其它".get(a_friend.raw.get(‘Sex‘, None), None)
    City = a_friend.raw.get(‘City‘, None)
    Province = a_friend.raw.get(‘Province‘, None)
    Signature = a_friend.raw.get(‘Signature‘, None)
    HeadImgUrl = a_friend.raw.get(‘HeadImgUrl‘, None)
    HeadImgFlag = a_friend.raw.get(‘HeadImgFlag‘, None)
    list_0 = [NickName, Sex, City, Province, Signature, HeadImgUrl, HeadImgFlag]
    lis.append(list_0)

 

输出 lis 

 

3.把列表存为 excel 表格，方便本地查看

 

#把列表转换为 xlsx 表格
def list_to_xlsx(filename, list):
    
    import openpyxl
　　#新建一个表格
    wb = openpyxl.Workbook()
    sheet = wb.active
    sheet.title = ‘Friends‘
    file_name = filename + ‘.xlsx‘
    for i in range(0, len(list)):
        for j in range(0, len(list[i])):
            sheet.cell(row = i+1, column = j+1, value = str(list[i][j]))
            
    wb.save(file_name)
    print("读写数据成功")

#把列表生成表格
list_to_xlsx(‘wechat_friend‘, lis)

 生成的表格为

 

4.对 execl 表格进行处理，利用表格信息生成词云

 

from wordcloud import WordCloud
import matplotlib.pyplot as plt
import pandas as pd
from pandas import read_excel

df = read_excel(‘wechat_friend.xlsx‘)

#使用 WordCloud 生成词云
word_list = df[‘City‘].fillna(‘‘).tolist()
new_text = ‘ ‘.join(word_list)
wordcloud = WordCloud(font_path=‘msyh.ttc‘, background_color = ‘white‘).generate(new_text)
plt.imshow(wordcloud)
plt.axis("off")
plt.show()
plt.savefig("WordCloud.png")
print("保存词云图")

 

词云图展示

5.利用表格信息生成网页词云图（即HTML）

 

#使用 pyecharts 生成词云
from pyecharts import WordCloud

city_list = df[‘City‘].fillna(‘‘).tolist()
count_city = pd.value_counts(city_list)
name = count_city.index.tolist()
value = count_city.tolist()

wordcloud = WordCloud(width=1300,height=620)
wordcloud.add("",name,value,word_size_range=[20,100])
#wordcloud.show_config()
#wordcloud.render(r‘D:\\wc.html‘)
wordcloud.render(‘wordcloud.html‘)
print("网页词云图已生成")

 词云图展示

 

6.利用表格信息生成好友地图分布图

 

#将好友展示在地图上 
from pyecharts import Map

province_list = df[‘Province‘].fillna(‘‘).tolist()
count_province = pd.value_counts(province_list)
attr = count_province.index.tolist()
value1 = count_province.tolist()
map = Map("各省微信好友分布", width=1200,height=600)
map.add("",attr, value1, maptype=‘china‘,is_visualmap=True,visualmap_text_color=‘#000‘,is_label_show=True)
#map.show_config()
map.render(‘map.html‘)
print("map已生成")

 

微信好友分布图展示

 

三.最后代码汇总

 把上面的代码，规范一下，可得——

# -*- coding: utf-8 -*-
"""
Created on Mon Jun  3 10:01:35 2019

@author: haiwe
"""
#获取微信接口 
from wxpy import *

#把获取的微信好友信息存为列表 
def wx_friend_information(friend_all):
    
    #先生成一个二维列表，用来储存好友信息 
    lis = [[‘NickName‘,‘Sex‘,‘City‘,‘Province‘,‘Signature‘,‘HeadImgUrl‘,‘HeadImgFlag‘]]
    #遍历 friend_all, 并使用 raw.get(‘ ‘, None), 来获取每一各朋友的基本信息 
    for a_friend in friend_all:
        NickName = a_friend.raw.get(‘NickName‘, None)
        Sex = 1:"男", 2:"女", 0:"其它".get(a_friend.raw.get(‘Sex‘, None), None)
        City = a_friend.raw.get(‘City‘, None)
        Province = a_friend.raw.get(‘Province‘, None)
        Signature = a_friend.raw.get(‘Signature‘, None)
        HeadImgUrl = a_friend.raw.get(‘HeadImgUrl‘, None)
        HeadImgFlag = a_friend.raw.get(‘HeadImgFlag‘, None)
        list_0 = [NickName, Sex, City, Province, Signature, HeadImgUrl, HeadImgFlag]
        lis.append(list_0)
        
    return lis

#把列表转换为 xlsx 表格，并保存表格       
def list_to_xlsx(filename, list):
    
    #使用 openpyxl 创建空表格 
    import openpyxl
    wb = openpyxl.Workbook()
    #获得当前正在显示的 sheet, 或 wb.get_active_sheet()  
    sheet = wb.active
    #表格的 sheet 页命名为 Friends 
    sheet.title = ‘Friends‘
    # execl 文件命名
    file_name = filename + ‘.xlsx‘
    #遍历 excel 表格，并读入数据 
    for i in range(0, len(list)):
        for j in range(0, len(list[i])):
            #为什么 i+1，j+1 不懂 
            sheet.cell(row = i+1, column = j+1, value = str(list[i][j]))
    #保存表格         
    wb.save(file_name)
    print("读写数据成功")
    
from wordcloud import WordCloud
import matplotlib.pyplot as plt
import pandas as pd
#使用 pandas 的 read_excel 模块读取execl 表格
from pandas import read_excel
#import numpy as np

#生成词云图 
def create_wordcloud(df):
    
    #把 df（dataframe） 中城市那列中 为空的 替换为 空字符，并把替换后的城市列存在 city_list中
    word_list = df[‘City‘].fillna(‘‘).tolist()
    #把 city_list 列表中的 各项 用空格隔开，并变为字符串 
    new_text = ‘ ‘.join(word_list)
    #设置词云图的字体为 ‘msyh.ttc‘, 背景为白色（默认为黑色），并用 generate() 生成词云图 
    wordcloud = WordCloud(font_path=‘msyh.ttc‘, background_color = ‘white‘).generate(new_text)
    #展示生成的词云图 
    plt.imshow(wordcloud)
    #plt.axis("off")
    #plt.show()
    plt.savefig("WordCloud.png")
    print("保存词云图")

#使用 pyecharts 生成词云    
def create_wordcloud_html(df):
    
    #调用 pyecharts 中 wordcloud 模块
    from pyecharts import WordCloud
    
    #把 df（dataframe） 中城市那列中 为空的 替换为 空字符，并把替换后的城市列存在 city_list中 
    city_list = df[‘City‘].fillna(‘‘).tolist()
    
    #使用 pandas.value_count() 返回 city_list 中各项和其对应的频数（两者一一对应），并存在 count_city 中  
    count_city = pd.value_counts(city_list)
    #把 count_city 中的项提取出来，并存在 name 中
    name = count_city.index.tolist()
    #把 count_city 中的频数提取出来 
    value = count_city.tolist()
    #生成词云图 
    wordcloud = WordCloud(width=1300,height=620)
    #往词云图中添加信息， 
    wordcloud.add("",name,value,word_size_range=[20,100])
    #输出词云图的配置 
    #wordcloud.show_config()
    #wordcloud.render(r‘D:\\wc.html‘)
    #保存词云图 
    wordcloud.render(‘wordcloud.html‘)
    print("网页词云图已生成")
    
    
#将好友展示在地图上
def create_map(df):
    
    #使用 pyechart 的 Map 模块，生成网页词云图
    from pyecharts import Map

    #把 df（dataframe） 中省份那列中 为空的 替换为 空字符，并把替换后的城市列存在 province_list中
    province_list = df[‘Province‘].fillna(‘‘).tolist()
    #使用 pandas.value_count() 返回 province_list 中各项和其对应的频数（两者一一对应），并存在 count_province 中
    count_province = pd.value_counts(province_list)
    #把 count_province 中的项提取出来，并存在 name 中
    attr = count_province.index.tolist()
    #把 count_province 中的频数提取出来
    value1 = count_province.tolist()
    #生成一个标题为 “ ”的空白 HTML 
    map = Map("各省微信好友分布", width=1200,height=600)
    #向空白 HTML 中添加 信息，设置地图为 中国地图，并显示地图，设置地图上文字颜色，并显示标签
    map.add("",attr, value1, maptype=‘china‘,is_visualmap=True,visualmap_text_color=‘#000‘,is_label_show=True)
    #map.show_config()
    #保存词云网页 
    map.render(‘map.html‘)
    print("map已生成")
    
    
if __name__ == "__main__":
    
    #获取登录二维码
    bot = Bot(cache_path = True)
    
    #获取微信朋友的基本数据
    friend_all = bot.friends()
    list = wx_friend_information(friend_all)
    list_to_xlsx(‘wechat_friend‘, list)
    
    df = read_excel(‘wechat_friend.xlsx‘)
    create_wordcloud(df)
    create_wordcloud_html(df)
    create_map(df)