用 python 進行微信好友信息分析
1. 使用到的庫
① wxpy:初始化微信機器人
② openpyxl:保存微信好友數據為Excel表格
③ pyecharts:生成可視化的地圖
④ wordcloud、matplotlib、jieba:生成詞云圖
【特別提醒】:pyecharts 庫用的是0.5.x版本,而在 pip 中安裝的為1.x.x版本,因此需要自行到【官網】中下載。
2. 基本功能
① 分析微信好友數據
② 生成詞云圖
③ 生成地圖展示
3. 代碼實現
此處使用類來實現
(1) 導入模塊
# 導入模塊from wxpy import Botimport openpyxlfrom pyecharts import Mapfrom wordcloud import WordCloudimport matplotlib.pyplot as pltimport 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:WindowsFontssimfang.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 Botimport openpyxlfrom pyecharts import Mapfrom wordcloud import WordCloudimport matplotlib.pyplot as pltimport jiebaclass 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:WindowsFontssimfang.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 微信信息分析的資料請關注好吧啦網其它相關文章!
相關文章: