用 python 进行微信好友信息分析
- 更新时间:2022-09-23 09:38:37
- 编辑:焦光济
参考资料
- 《Python趣味编程入门》代码文件 配套资源 / 34.34 KB / (英)迈克·桑德斯(Mike Saund 推荐度:
- 《Python自动化测试入门与进阶实战》源代码 配套资源 / 126 MB / 唐文 推荐度:
- 21天学通Python PDF 电子书 / 58.3 MB / 刘凌霞,郝宁波,吴海涛 推荐度:
正文内容
python 微信信息分析
1. 使用到的库
① wxpy:初始化微信机器人
② openpyxl:保存微信好友数据为Excel表格
③ pyecharts:生成可视化的地图
④ wordcloud、matplotlib、jieba:生成词云图
【特别提醒】:pyecharts 库用的是0.5.x版本,而在 pip 中安装的为1.x.x版本,因此需要自行到【官网】中下载。
2. 基本功能
① 分析微信好友数据
② 生成词云图
③ 生成地图展示
3. 代码实现
此处使用类来实现
(1) 导入模块
# 导入模块 from wxpy import Bot import openpyxl from pyecharts import Map from wordcloud import WordCloud import matplotlib.pyplot as plt import jieba
(2) 初始化机器人和获取微信好友的源信息
此处调用 Bot() 方法,需要扫码登陆微信网页版,后续操作才能进行。
def __init__(self, ToExcelFile="", ToCityFile="", ToMapProvinceFile="", ToMapCityFile=""): ''' 初始化机器人和其他参数 ''' # 初始化机器人,需要扫码 self.bot = Bot() # 获取我所有的微信好友信息 - 存储基础信息(未处理) self.allFriends_Info = self.bot.friends() # 我的微信好友个数 self.allFriends_Num = len(self.allFriends_Info) # 保存微信好友信息的表格文件路径(.xlsx) self.ExcelFile = ToExcelFile # 保存城市词云图的文件路径(.png/.jpg) self.WCOfCityFile = ToCityFile # 保存省份地图的文件路径(.html) self.MapProvinceFile = ToMapProvinceFile # 其他可用参数 self.MapCityFile = ToMapCityFile # 自动调用run方法,使得在实例化对象后自动运行其他函数 self.run()
(3) 统计和处理微信好友的信息
除了列出的还有 个性签名、头像等其他属性。
def getFriendsInfo(self): ''' 获取微信好友的全部信息 ''' # 存储微信好友的信息(经过信息处理的) self.friendsInfo = [] # 定义列标题 self.infoTitle = ['NickName', 'RemarkName', 'Sex', 'Province', 'City'] for aFriend in self.allFriends_Info: # 获取昵称 NickName = aFriend.raw.get(self.infoTitle[0], None) # 获取备注 RemarkName = aFriend.raw.get(self.infoTitle[1], None) # 获取性别 Sex = {1:"男", 2:"女", 0:"其他"}.get(aFriend.raw.get(self.infoTitle[2], None), None) # 获取省份 Province = aFriend.raw.get(self.infoTitle[3], None) # 获取城市 City = aFriend.raw.get(self.infoTitle[4], None) lisTmp = [NickName, RemarkName, Sex, Province, City] self.friendsInfo.append(lisTmp)
(4) 保存微信好友的信息
在这保存为Excel表格,在代码中插入表头行,为了便于阅读。
def saveFriendsInfoAsExcel(self, ExcelName): ''' 保存微信好友的信息到 Excel 表格中 ''' # 生成openpyxl对象 workbook = openpyxl.Workbook() # 激活表格 sheet = workbook.active # 设置表格标题 sheet.title = 'WeChatFriendsInfo' # 填充列标题到第一行 for _ in range(len(self.infoTitle)): sheet.cell(row=1, column=_+1, value=self.infoTitle[_]) # 填充微信好友信息,从第二行开始 for i in range(self.allFriends_Num): for j in range(len(self.infoTitle)): sheet.cell(row=i+2, column=j+1, value=str(self.friendsInfo[i][j])) # 若文件名非空,则保存到该路径下 if ExcelName != "": workbook.save(ExcelName) print(">>> Save WeChat friends' information successfully!")
(5) 分析微信好友的信息
def quiteAnalyzeFriendsInfo(self): ''' 分析数据,一步到位,直接了当 ''' print(self.allFriends_Info.stats_text())
(6) 生成city词云图
def creatWordCloudOfCity(self, CityName): ''' 使用获取的数据生成city词云图 ''' # 获取所有的城市 cityStr = "" for i in range(self.allFriends_Num): if self.friendsInfo[i][4] not in cityStr: cityStr += " " + self.friendsInfo[i][4] #jieba库精确模式分词 wordlist = jieba.lcut(cityStr) cityStr = ' '.join(wordlist) # 加载背景图片 #cloud_mask = np.array(Image.open(BackGroundFile)) #设置词云图属性 font = r'C:\Windows\Fonts\simfang.ttf' # 设置字体路径 wc = WordCloud( background_color = 'black', # 背景颜色 #mask = cloud_mask, # 背景图片 max_words = 100, # 设置最大显示的词云数 font_path = font, # 设置字体形式(在本机系统中) height = 300, # 图片高度 width = 600, # 图片宽度 max_font_size = 100, # 字体最大值 random_state = 100, # 配色方案的种类 ) # 生成词云图 myword = wc.generate(cityStr) #展示词云图 plt.imshow(myword) plt.axis('off') plt.show() # 若文件名非空,则保存到该路径下 if CityName != "": #保存词云图 wc.to_file(CityName) print(">>> Creat WeChat wordcloud of city successfully!")
(7) 生成province地图
def creatMapProvince(self, MapFile): ''' 使用获取的数据生成province地图 ''' # 获取所有省份 provinceList, provinceNum = [], [] for i in range(self.allFriends_Num): if self.friendsInfo[i][3] not in provinceList: provinceList.append(self.friendsInfo[i][3]) provinceNum.append(0) for i in range(self.allFriends_Num): for j in range(len(provinceList)): if self.friendsInfo[i][3] == provinceList[j]: provinceNum[j] += 1 # 生成 Map map = Map("各省微信好友分布", width=1000, height=800) map.add("", provinceList, provinceNum, maptype="china", is_visualmap=True, visual_text_color='#000') # 若文件名非空,则保存到该路径下 if MapFile != "": map.render(MapFile) print(">>> Creat WeChat Map of Provinces seccessfully!")
(8) 生成city地图
def creatMapCity(self, MapFile): ''' 使用获取的数据生成city地图 ''' # 获取所有省份 CityList, CityNum = [], [] for i in range(self.allFriends_Num): if self.friendsInfo[i][4] not in CityList: CityList.append(self.friendsInfo[i][4]) CityNum.append(0) for i in range(self.allFriends_Num): for j in range(len(CityList)): if self.friendsInfo[i][4] == CityList[j]: CityNum[j] += 1 for i in range(len(CityList)): CityList[i] += '市' # 生成 Map map = Map("各市微信好友分布", width=1000, height=800) map.add("", CityList, CityNum, maptype="广东", is_visualmap=True, visual_text_color='#000') # 若文件名非空,则保存到该路径下 if MapFile != "": map.render(MapFile) print(">>> Creat WeChat Map of Cities seccessfully!")
有了上述实现各个功能的方法,那么就差一个调用各种方法的方法了。
(9) run方法
def run(self): # 获取微信好友信息 self.getFriendsInfo() print(">>> Get WeChat friends' information successfully!") print(">>> Members:", self.allFriends_Num) # 保存微信好友信息 self.saveFriendsInfoAsExcel(self.ExcelFile) # 分析微信好友信息 self.quiteAnalyzeFriendsInfo() # 使用微信好友的 city 产生词云图 self.creatWordCloudOfCity(self.WCOfCityFile) # 生成微信好友的 province 地图 self.creatMapProvince(self.MapProvinceFile) # 生成微信好友的 city 地图 self.creatMapCity(self.MapCityFile)
对于文件路径,在main函数中传递即可。【注】:上述代码都在类中,在此处结束,下面为main函数
if __name__ == "__main__": ToExcelFile = "./WeChatAnalyze//FriendsInfo.xlsx" # 微信好友信息的Excel表格保存路径 ToPictureFile = "./WeChatAnalyze//CityWordCloud.png" # 微信好友信息city词云图保存路径 ToMapFileProvince = "./WeChatAnalyze//WeChatProvinceMap.html" # 微信好友信息province地图保存路径 ToMapFileCity = "./WeChatAnalyze//WeChatCityMap.html" # 微信好友信息city地图保存路径 # WeChatRobot对象实例化 robot = WeChatRobot(ToExcelFile, ToPictureFile, ToMapFileProvince, ToMapFileCity)
是不是觉得Main函数很简短,哈哈,没错,就是这么简!
接下来看看实现的效果吧!
>>> 这个是终端显示效果
>>> 这个是保存为Excel表格的内容
>>> 这个是微信好友各省的分布
>>> 这个是微信好友各市的分布
完整代码
# -*- coding: utf-8 -*- ''' This is a program which can analyze datas of WeChat friends. @author: bpf ''' # 导入模块 from wxpy import Bot import openpyxl from pyecharts import Map from wordcloud import WordCloud import matplotlib.pyplot as plt import jieba class WeChatRobot: '''====================== 1. 获取微信好友信息 ======================''' def __init__(self, ToExcelFile="", ToCityFile="", ToMapProvinceFile="", ToMapCityFile=""): ''' 初始化机器人和其他参数 ''' # 初始化机器人,需要扫码 self.bot = Bot() # 获取我所有的微信好友信息 - 存储基础信息(未处理) self.allFriends_Info = self.bot.friends() # 我的微信好友个数 self.allFriends_Num = len(self.allFriends_Info) # 保存微信好友信息的表格文件路径(.xlsx) self.ExcelFile = ToExcelFile # 保存城市词云图的文件路径(.png/.jpg) self.WCOfCityFile = ToCityFile # 保存省份地图的文件路径(.html) self.MapProvinceFile = ToMapProvinceFile # 其他可用参数 self.MapCityFile = ToMapCityFile # 自动调用run方法,使得在实例化对象后自动运行其他函数 self.run() '''====================== 2. 统计微信好友信息 ======================''' def getFriendsInfo(self): ''' 获取微信好友的全部信息 ''' # 存储微信好友的信息(经过信息处理的) self.friendsInfo = [] # 定义列标题 self.infoTitle = ['NickName', 'RemarkName', 'Sex', 'Province', 'City'] for aFriend in self.allFriends_Info: # 获取昵称 NickName = aFriend.raw.get(self.infoTitle[0], None) # 获取备注 RemarkName = aFriend.raw.get(self.infoTitle[1], None) # 获取性别 Sex = {1:"男", 2:"女", 0:"其他"}.get(aFriend.raw.get(self.infoTitle[2], None), None) # 获取省份 Province = aFriend.raw.get(self.infoTitle[3], None) # 获取城市 City = aFriend.raw.get(self.infoTitle[4], None) lisTmp = [NickName, RemarkName, Sex, Province, City] self.friendsInfo.append(lisTmp) '''====================== 3. 保存微信好友信息 ======================''' def saveFriendsInfoAsExcel(self, ExcelName): ''' 保存微信好友的信息到 Excel 表格中 ''' # 生成openpyxl对象 workbook = openpyxl.Workbook() # 激活表格 sheet = workbook.active # 设置表格标题 sheet.title = 'WeChatFriendsInfo' # 填充列标题到第一行 for _ in range(len(self.infoTitle)): sheet.cell(row=1, column=_+1, value=self.infoTitle[_]) # 填充微信好友信息,从第二行开始 for i in range(self.allFriends_Num): for j in range(len(self.infoTitle)): sheet.cell(row=i+2, column=j+1, value=str(self.friendsInfo[i][j])) # 若文件名非空,则保存到该路径下 if ExcelName != "": workbook.save(ExcelName) print(">>> Save WeChat friends' information successfully!") '''====================== 4. 分析微信好友信息 ======================''' def quiteAnalyzeFriendsInfo(self): ''' 分析数据,一步到位,直接了当 ''' print(self.allFriends_Info.stats_text()) '''====================== 5. 产生city词云图 ======================''' def creatWordCloudOfCity(self, CityName): ''' 使用获取的数据生成city词云图 ''' # 获取所有的城市 cityStr = "" for i in range(self.allFriends_Num): if self.friendsInfo[i][4] not in cityStr: cityStr += " " + self.friendsInfo[i][4] #jieba库精确模式分词 wordlist = jieba.lcut(cityStr) cityStr = ' '.join(wordlist) # 加载背景图片 #cloud_mask = np.array(Image.open(BackGroundFile)) #设置词云图属性 font = r'C:\Windows\Fonts\simfang.ttf' # 设置字体路径 wc = WordCloud( background_color = 'black', # 背景颜色 #mask = cloud_mask, # 背景图片 max_words = 100, # 设置最大显示的词云数 font_path = font, # 设置字体形式(在本机系统中) height = 300, # 图片高度 width = 600, # 图片宽度 max_font_size = 100, # 字体最大值 random_state = 100, # 配色方案的种类 ) # 生成词云图 myword = wc.generate(cityStr) #展示词云图 plt.imshow(myword) plt.axis('off') plt.show() # 若文件名非空,则保存到该路径下 if CityName != "": #保存词云图 wc.to_file(CityName) print(">>> Creat WeChat wordcloud of city successfully!") '''===================== 6. 产生province地图 =====================''' def creatMapProvince(self, MapFile): ''' 使用获取的数据生成province地图 ''' # 获取所有省份 provinceList, provinceNum = [], [] for i in range(self.allFriends_Num): if self.friendsInfo[i][3] not in provinceList: provinceList.append(self.friendsInfo[i][3]) provinceNum.append(0) for i in range(self.allFriends_Num): for j in range(len(provinceList)): if self.friendsInfo[i][3] == provinceList[j]: provinceNum[j] += 1 # 生成 Map map = Map("各省微信好友分布", width=1000, height=800) map.add("", provinceList, provinceNum, maptype="china", is_visualmap=True, visual_text_color='#000') # 若文件名非空,则保存到该路径下 if MapFile != "": #map.show_config() map.render(MapFile) print(">>> Creat WeChat Map of Provinces seccessfully!") '''===================== 7. 产生city地图 =====================''' def creatMapCity(self, MapFile): ''' 使用获取的数据生成city地图 ''' # 获取所有省份 CityList, CityNum = [], [] for i in range(self.allFriends_Num): if self.friendsInfo[i][4] not in CityList: CityList.append(self.friendsInfo[i][4]) CityNum.append(0) for i in range(self.allFriends_Num): for j in range(len(CityList)): if self.friendsInfo[i][4] == CityList[j]: CityNum[j] += 1 for i in range(len(CityList)): CityList[i] += '市' # 生成 Map map = Map("各市微信好友分布", width=1000, height=800) map.add("", CityList, CityNum, maptype="广东", is_visualmap=True, visual_text_color='#000') # 若文件名非空,则保存到该路径下 if MapFile != "": map.render(MapFile) print(">>> Creat WeChat Map of Cities seccessfully!") '''===================== 8. 自动执行函数 =====================''' def run(self): # 获取微信好友信息 self.getFriendsInfo() print(">>> Get WeChat friends' information successfully!") print(">>> Members:", self.allFriends_Num) # 保存微信好友信息 self.saveFriendsInfoAsExcel(self.ExcelFile) # 分析微信好友信息 self.quiteAnalyzeFriendsInfo() # 使用微信好友的 city 产生词云图 self.creatWordCloudOfCity(self.WCOfCityFile) # 生成微信好友的 province 地图 self.creatMapProvince(self.MapProvinceFile) # 生成微信好友的 city 地图 self.creatMapCity(self.MapCityFile) if __name__ == "__main__": ToExcelFile = "./WeChatAnalyze//FriendsInfo.xlsx" # 微信好友信息的Excel表格保存路径 ToPictureFile = "./WeChatAnalyze//CityWordCloud.png" # 微信好友信息city词云图保存路径 ToMapFileProvince = "./WeChatAnalyze//WeChatProvinceMap.html" # 微信好友信息province地图保存路径 ToMapFileCity = "./WeChatAnalyze//WeChatCityMap.html" # 微信好友信息city地图保存路径 # WeChatRobot对象实例化 robot = WeChatRobot(ToExcelFile, ToPictureFile, ToMapFileProvince, ToMapFileCity)
以上就是用 python 进行微信好友信息分析的详细内容,更多关于python 微信信息分析的资料请关注码农之家其它相关文章!
python相关教程
-
python整数反转输出的实例讲解
给定一个32位的符号整数,返回它的反转整数,正常整数方法实现,利用余数*10累加的方法完成。需要注意的是,python对整数除法采用“向下取整”机制,所以正数和负数要区别运算。整数转字
发布时间:2021-08-25
-
精选Python开发工程师面试题
你知道吗?实际上Python早在20世纪90年代初就已经诞生,可是火爆时间却并不长,就小编本人来说,也是前几年才了解到它。据统计,目前Python开发人员的薪资待遇为10K以上,这样的诱惑很难让人拒
发布时间:2019-11-06
-
python协程深入详细分析
这篇文章主要给大家介绍了关于python中协程的相关资料,文中通过示例代码介绍的非常详细,对大家学习或者使用python具有一定的参考学习价值,需要的朋友们下面来一起学习学习吧
发布时间:2020-07-17
-
基于python实现简单C/S模式代码实例
这篇文章主要介绍了基于python实现简单C/S模式代码实例,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
发布时间:2021-04-16
-
python map的实例用法讲解
map函数的原型是map(function, iterable, …),它的返回结果是一个列表。下面这篇文章主要给大家介绍了关于python中map的基本用法,文中通过示例代码介绍的非常详细,需要的朋友可以参考下
发布时间:2019-08-30
-
python表示根号运算的方法
Python中我们可以使用指数运算符**来计算一个数的平方根,该方法只适用于正数。负数和复数可以使用sqrt()函数进行运算。
发布时间:2019-07-02
-
python如何通过函数名的字符串来调用这个函数
给大家整理了关于python的教程,通过字符串名称调用函数,可使用eval()方法,它执行一个字符串表达式并返回表达式的值;locals()和 globals()两个内置函数,通过它们可以访问局部和全局变量。
发布时间:2022-06-17
-
python语法实现单向循环链表
之前的学习中都是用C语言编写的链表实现,今天小编将带大家一起来学习用python编写单向循环链表。
发布时间:2020-01-18
-
python3使用matplotlib绘制散点图实例详解
这篇文章主要为大家详细介绍了python3使用matplotlib绘制散点图,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
发布时间:2020-01-28
-
python爬虫数据解析之正则表达式
正则表达式是一个特殊的字符序列,它能帮助你方便的检查一个字符串是否与某种模式匹配。就是 事先定义好的一些特定字符、及这些特定字符的组合,组成一个“规则字符”,这个“规则字
发布时间:2020-01-28