我的第一个微信好友分析

主要介绍利用网页端微信获取数据,实现个人微信好友数据的获取,并进行一些简单的数据分析。

一、所需要的七个第三方库及其安装

1、Pillow

PIL:Python Imaging Library,已经是 Python 平台事实上的图像处理标准库。PIL功能非常强大,但API却非常简单易用。

如果安装了Anaconda,Pillow就已经可用了。否则,需要在命令行下通过pip安装:

pip install pillow

如果遇到Permission denied安装失败,请加上sudo重试。

2、Pyecharts

是一个用于生成 Echarts 图表的类库。Echarts 是百度开源的一个数据可视化 JS 库。用 Echarts 生成的图可视化效果非常棒,pyecharts 是为了与 Python 进行对接,方便在 Python 中直接使用数据生成图。使用pyecharts可以生成独立的网页,也可以在flask、django中集成使用。

可在命令行通过pip安装:

pip install pyecharts 

3、Itchat

itchat 是一个开源的微信个人号接口,使用 python 调用微信从未如此简单。

可在在命令行通过pip安装:

pip install itchat

4、Jieba

Jieba库是一款优秀的 Python 第三方中文分词库,jieba 支持三种分词模式:精确模式、全模式和搜索引擎模式,下面是三种模式的特点。

精确模式:试图将语句最精确的切分,不存在冗余数据,适合做文本分析

全模式:将语句中所有可能是词的词语都切分出来,速度很快,但是存在冗余数据

搜索引擎模式:在精确模式的基础上,对长词再次进行切分

可在在命令行通过pip安装:

pip install jieba

5、Numpy

NumPy 是一个 Python 的第三方库,代表 “Numeric Python”,主要用于数学/科学计算。 它是一个由多维数组对象和用于处理数组的例程集合组成的库。

使用 Numpy 我们可以轻松进行如下等计算:

  • 数组的算数和逻辑运算。
  • 傅立叶变换和用于图形操作的例程。
  • 与线性代数有关的操作。 NumPy 拥有线性代数和随机数生成的内置函数。

可在在命令行通过pip安装:

pip install Numpy

6、Pandas

Pandas 是基于 NumPy 的一种工具,该工具是为了解决数据分析任务而创建的。Pandas纳入了大量库和一些标准的数据模型,提供了高效地操作大型数据集所需的工具。pandas 提

供了大量能使我们快速便捷地处理数据的函数和方法。你很快就会发现,它是使 Python 成为强大而高效的数据分析环境的重要因素之一。

可在在命令行通过pip安装:

pip install Pandas

7、wxpy

wxpy 在 itchat 的基础上,通过大量接口优化提升了模块的易用性,并进行丰富的功能扩展。

可在在命令行通过pip安装:

pip install wxpy

8、安装地图数据包:

pip install echarts-china-provinces-pypkg
pip install echarts-countries-pypkg

二、开始分析

1、首先登录自己的微信

from wxpy import *
#初始化机器人,选择缓存模式(扫码)登录
bot = Bot(cache_path=True)
#获取我的所有微信好友信息
friend_all = bot.friends()

运行登录代码会自动弹出一个二维码页面,打开如下图 1, 用手机扫码同意后,进入微信并获取微信好友的相关信息。

 

print(friend_all[0].raw)#获取自己的全部信息
print(len(friend_all))#输出自己的微信好友人数

为了获取好友信息中需要的部分,我们对信息需要进行处理。从上面的获取信息全字段来看,我们获取的每位好友的信息都是一个字典,字典里只有'City'、'Province'、'Signature'、
'NickName'、'HeadImgUrl'、'Sex'是我们需要的。下面我们就对这几个 key 进行提取。

lis=[]
infoTitle=['NickName', 'RemarkName', 'Sex', 'Province', 'City']
def getmessage():                       #获取好友的全部信息
    for a_friend in friend_all:
        NickName = a_friend.raw.get(infoTitle[0], None) #昵称
        RemarkName = a_friend.raw.get(infoTitle[1], None)#备注
        Sex = {1:"男", 2:"女", 0:"保密"}.get(a_friend.raw.get(infoTitle[2], None), None)#性别
        Province = a_friend.raw.get(infoTitle[3], None)#省份
        City = a_friend.raw.get(infoTitle[4], None)#城市
        list_0 = [NickName, RemarkName, Sex, Province, City]
        lis.append(list_0)

 

 将 lis 列表能够保存到 excel 中:

def savemessage():
    workbook = openpyxl.Workbook()
    sheet = workbook.active
    sheet.title = '微信好友信息'
    for k in range(len(infoTitle)):
        sheet.cell(row=1, column=k+1, value=infoTitle[k])
    for i in range(FriNum):
        for j in range(len(infoTitle)):
            sheet.cell(row=i+2, column=j+1, value=str(lis[i][j]))
    workbook.save('C:\\Users\\Administrator\\Desktop\\wxfx.xlsx')

运行后结果如图:

 

对数据进行简单的分析:

Friends = bot.friends()
data = Friends.stats_text(total=True, sex=True,top_provinces=30, top_cities=500)
print(data)

结果如图:

 

将好友所在的城市用词云图显示:

def creatcity():
    cityStr = ""
    for i in range(FriNum):
        if lis[i][4] not in cityStr:
            cityStr += " " +lis[i][4]
    wordlist = jieba.lcut(cityStr)
    cityStr = ' '.join(wordlist)
    wc = WordCloud(
        background_color = 'white',
        max_words = 50,
        font_path = 'simhei.ttf',
        height = 400,
        width = 400,
        max_font_size = 100,
        random_state = 50,
        )
    myword = wc.generate(cityStr)
    plt.imshow(myword)
    plt.axis('off')
    plt.show()
    wc.to_file('C:\\Users\\Administrator\\Desktop\\city.jpg')

运行结果如图:

 

 

 

生成微信好友在广东省分布的城市图:

def creatcitymap():
    CityList=[]
    CityNum = []
    for i in range(FriNum):
        if lis[i][4] not in CityList:
            CityList.append(lis[i][4])
            CityNum.append(0)
    for i in range(FriNum):
        for j in range(len(CityList)):
            if lis[i][4] == CityList[j]:
                CityNum[j] += 1
    for i in range(len(CityList)):
        CityList[i] += ''
    map=Map(width=1000, height=1000)
    map.add("", CityList, CityNum, maptype="广东", is_visualmap=True, visual_text_color='#000')
    map.render('C:\\Users\\Administrator\\Desktop\\citymap.html')

结果如图:

 

 这就是对微信的一些简单的数据分析。

 

 

完整代码如下:

from wxpy import Bot
import openpyxl
from pyecharts import Map
from wordcloud import WordCloud
import matplotlib.pyplot as plt
import jieba
#初始化机器人,选择缓存模式(扫码)登录
bot = Bot(cache_path=True)
#获取我的所有微信好友信息
friend_all = bot.friends()                                  #self.allFriends_Info = self.bot.friends()
FriNum=len(friend_all)                                      #self.allFriends_Num = len(self.allFriends_Info)
lis=[]
infoTitle=['NickName', 'RemarkName', 'Sex', 'Province', 'City']
def getmessage():                       #获取好友的全部信息
    for a_friend in friend_all:
        NickName = a_friend.raw.get(infoTitle[0], None) #昵称
        RemarkName = a_friend.raw.get(infoTitle[1], None)#备注
        Sex = {1:"", 2:"", 0:"保密"}.get(a_friend.raw.get(infoTitle[2], None), None)#性别
        Province = a_friend.raw.get(infoTitle[3], None)#省份
        City = a_friend.raw.get(infoTitle[4], None)#城市
        list_0 = [NickName, RemarkName, Sex, Province, City]
        lis.append(list_0)
def savemessage():
    workbook = openpyxl.Workbook()
    sheet = workbook.active
    sheet.title = '微信好友信息'
    for k in range(len(infoTitle)):
        sheet.cell(row=1, column=k+1, value=infoTitle[k])
    for i in range(FriNum):
        for j in range(len(infoTitle)):
            sheet.cell(row=i+2, column=j+1, value=str(lis[i][j]))
    workbook.save('C:\\Users\\Administrator\\Desktop\\wxfx.xlsx')


def creatcity():
    cityStr = ""
    for i in range(FriNum):
        if lis[i][4] not in cityStr:
            cityStr += " " +lis[i][4]
    wordlist = jieba.lcut(cityStr)
    cityStr = ' '.join(wordlist)
    wc = WordCloud(
        background_color = 'white',
        max_words = 50,
        font_path = 'simhei.ttf',
        height = 400,
        width = 400,
        max_font_size = 100,
        random_state = 50,
        )
    myword = wc.generate(cityStr)
    plt.imshow(myword)
    plt.axis('off')
    plt.show()
    wc.to_file('C:\\Users\\Administrator\\Desktop\\city.jpg')

def creatcitymap():
    CityList=[]
    CityNum = []
    for i in range(FriNum):
        if lis[i][4] not in CityList:
            CityList.append(lis[i][4])
            CityNum.append(0)
    for i in range(FriNum):
        for j in range(len(CityList)):
            if lis[i][4] == CityList[j]:
                CityNum[j] += 1
    for i in range(len(CityList)):
        CityList[i] += ''
    map=Map(width=1000, height=1000)
    map.add("", CityList, CityNum, maptype="广东", is_visualmap=True, visual_text_color='#000')
    map.render('C:\\Users\\Administrator\\Desktop\\citymap.html')

def main():
    getmessage()
    savemessage()
    creatcity()
    creatcitymap()
main()

 

posted on 2019-06-02 09:48  芷若丶初见  阅读(270)  评论(0编辑  收藏  举报