用 python 进行微信好友信息分析
作者:步平凡 时间:2022-03-29 15:52:44
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)
来源:https://www.cnblogs.com/bpf-1024/p/10962332.html
标签:python,微信
0
投稿
猜你喜欢
教你使用Python画圣诞树做浪漫的程序员
2023-03-19 21:39:02
Python Socketserver实现FTP文件上传下载代码实例
2022-10-10 11:03:04
python利用numpy存取文件案例教程
2023-05-22 03:55:53
Python数组定义方法
2021-05-30 08:46:39
简单了解python变量的作用域
2022-07-21 13:10:08
Oracle AS关键字 提示错误
2011-04-18 12:42:00
Python3 Post登录并且保存cookie登录其他页面的方法
2023-08-18 22:45:52
python实现在pandas.DataFrame添加一行
2022-04-14 01:53:23
Python列表排序方法reverse、sort、sorted详解
2022-01-06 22:16:39
python获取指定目录下所有文件名列表的方法
2022-03-24 08:05:39
Go 语言的 :=的具体使用
2024-05-10 13:58:56
Python线程threading模块用法详解
2023-01-30 03:55:01
详解Python给照片换底色(蓝底换红底)
2023-02-03 02:07:45
SQL Serve数据库到DB2连接服务器的实现过程全解
2010-08-31 15:06:00
利用Python实现绘制3D爱心的代码分享
2021-03-30 23:25:12
innerHTML 的些摘记
2009-06-01 15:42:00
JS实现键值对遍历json数组功能示例
2024-04-10 10:52:40
asp函数解决SQL注入漏洞
2008-10-12 19:53:00
JavaScript实现获取最近7天的日期的方法详解
2024-04-27 15:20:59
使用Python多线程爬虫爬取电影天堂资源
2022-12-06 11:56:27