使用 python 進行微信好友分析

使用 python 進行微信好友分析

1. 使用到的庫

① wxpy：初始化微信機器人

② openpyxl：保存微信好友數據爲Excel表格

③ pyecharts：生成可視化的地圖

④ wordcloud、matplotlib、jieba：生成詞雲圖

【特別提醒】：pyecharts 庫用的是0.5.x版本，而在 pip 中安裝的爲1.x.x版本，所以須要自行到【官網】中下載。

2. 基本功能

① 分析微信好友數據

② 生成詞雲圖

③ 生成地圖展現

3. 代碼實現

此處使用類來實現

(1) 導入模塊

1 #  導入模塊
2 from wxpy import Bot
3 import openpyxl
4 from pyecharts import Map
5 from wordcloud import WordCloud
6 import matplotlib.pyplot as plt
7 import jieba

 

(2) 初始化機器人和獲取微信好友的源信息

此處調用 Bot() 方法，須要掃碼登錄微信網頁版，後續操做才能進行。

 1 def __init__(self, ToExcelFile="", ToCityFile="", ToMapProvinceFile="", ToMapCityFile=""):
 2     ''' 初始化機器人和其餘參數 '''
 3     # 初始化機器人，須要掃碼
 4     self.bot = Bot()
 5     # 獲取我全部的微信好友信息 - 存儲基礎信息(未處理)
 6     self.allFriends_Info = self.bot.friends()
 7     # 個人微信好友個數
 8     self.allFriends_Num = len(self.allFriends_Info)
 9     # 保存微信好友信息的表格文件路徑(.xlsx)
10     self.ExcelFile = ToExcelFile
11     # 保存城市詞雲圖的文件路徑(.png/.jpg)
12     self.WCOfCityFile = ToCityFile
13     # 保存省份地圖的文件路徑(.html)
14     self.MapProvinceFile = ToMapProvinceFile
15     # 其餘可用參數
16     self.MapCityFile = ToMapCityFile
17     # 自動調用run方法，使得在實例化對象後自動運行其餘函數
18     self.run()

 

(3) 統計和處理微信好友的信息

除了列出的還有 個性簽名、頭像等其餘屬性。

 1 def getFriendsInfo(self):
 2     ''' 獲取微信好友的所有信息 '''
 3     # 存儲微信好友的信息(通過信息處理的)
 4     self.friendsInfo = []
 5     # 定義列標題
 6     self.infoTitle = ['NickName', 'RemarkName', 'Sex', 'Province', 'City']
 7     for aFriend in self.allFriends_Info:
 8         # 獲取暱稱
 9         NickName = aFriend.raw.get(self.infoTitle[0], None)
10         # 獲取備註
11         RemarkName = aFriend.raw.get(self.infoTitle[1], None)
12         # 獲取性別
13         Sex = {1:"男", 2:"女", 0:"其餘"}.get(aFriend.raw.get(self.infoTitle[2], None), None)
14         # 獲取省份
15         Province = aFriend.raw.get(self.infoTitle[3], None)
16         # 獲取城市
17         City = aFriend.raw.get(self.infoTitle[4], None)
18         lisTmp = [NickName, RemarkName, Sex, Province, City]
19         self.friendsInfo.append(lisTmp)

 

(4) 保存微信好友的信息

在這保存爲Excel表格，在代碼中插入表頭行，爲了便於閱讀。

 1 def saveFriendsInfoAsExcel(self, ExcelName):
 2     ''' 保存微信好友的信息到 Excel 表格中 '''
 3     # 生成openpyxl對象
 4     workbook = openpyxl.Workbook()
 5     # 激活表格
 6     sheet = workbook.active
 7     # 設置表格標題
 8     sheet.title = 'WeChatFriendsInfo'
 9     # 填充列標題到第一行
10     for _ in range(len(self.infoTitle)):
11         sheet.cell(row=1, column=_+1, value=self.infoTitle[_])
12     # 填充微信好友信息，從第二行開始
13     for i in range(self.allFriends_Num):
14         for j in range(len(self.infoTitle)):
15             sheet.cell(row=i+2, column=j+1, value=str(self.friendsInfo[i][j]))
16     # 若文件名非空，則保存到該路徑下
17     if ExcelName != "":
18         workbook.save(ExcelName)
19         print(">>> Save WeChat friends' information successfully!")

 

(5) 分析微信好友的信息

1 def quiteAnalyzeFriendsInfo(self):
2     ''' 分析數據，一步到位，直接了當 '''
3     print(self.allFriends_Info.stats_text())

 

(6) 生成city詞雲圖

 1 def creatWordCloudOfCity(self, CityName):
 2     ''' 使用獲取的數據生成city詞雲圖 '''
 3     # 獲取全部的城市
 4     cityStr = ""
 5     for i in range(self.allFriends_Num):
 6         if self.friendsInfo[i][4] not in cityStr:
 7             cityStr += " " + self.friendsInfo[i][4]
 8     #jieba庫精確模式分詞
 9     wordlist = jieba.lcut(cityStr)
10     cityStr = ' '.join(wordlist)
11     # 加載背景圖片
12     #cloud_mask = np.array(Image.open(BackGroundFile))
13     #設置詞雲圖屬性
14     font = r'C:\Windows\Fonts\simfang.ttf' # 設置字體路徑
15     wc = WordCloud(
16         background_color = 'black',     # 背景顏色
17         #mask = cloud_mask,             # 背景圖片
18         max_words = 100,                # 設置最大顯示的詞雲數
19         font_path = font,               # 設置字體形式（在本機系統中）
20         height = 300,                   # 圖片高度
21         width = 600,                    # 圖片寬度
22         max_font_size = 100,            # 字體最大值
23         random_state = 100,             # 配色方案的種類
24         )
25     # 生成詞雲圖
26     myword = wc.generate(cityStr)
27     #展現詞雲圖
28     plt.imshow(myword)
29     plt.axis('off')
30     plt.show()
31     # 若文件名非空，則保存到該路徑下
32     if CityName != "":
33         #保存詞雲圖
34         wc.to_file(CityName)
35         print(">>> Creat WeChat wordcloud of city successfully!")

 

(7) 生成province地圖

 1 def creatMapProvince(self, MapFile):
 2     ''' 使用獲取的數據生成province地圖 '''
 3     # 獲取全部省份
 4     provinceList, provinceNum = [], []
 5     for i in range(self.allFriends_Num):
 6         if self.friendsInfo[i][3] not in provinceList:
 7             provinceList.append(self.friendsInfo[i][3])
 8             provinceNum.append(0)
 9     for i in range(self.allFriends_Num):
10         for j in range(len(provinceList)):
11             if self.friendsInfo[i][3] == provinceList[j]:
12                 provinceNum[j] += 1
13     # 生成 Map
14     map = Map("各省微信好友分佈", width=1000, height=800)
15     map.add("", provinceList, provinceNum, maptype="china", is_visualmap=True, visual_text_color='#000')
16     # 若文件名非空，則保存到該路徑下
17     if MapFile != "":
18         map.render(MapFile)
19         print(">>> Creat WeChat Map of Provinces seccessfully!")

 

(8) 生成city地圖

 1 def creatMapCity(self, MapFile):
 2         ''' 使用獲取的數據生成city地圖 '''
 3         # 獲取全部省份
 4         CityList, CityNum = [], []
 5         for i in range(self.allFriends_Num):
 6             if self.friendsInfo[i][4] not in CityList:
 7                 CityList.append(self.friendsInfo[i][4])
 8                 CityNum.append(0)
 9         for i in range(self.allFriends_Num):
10             for j in range(len(CityList)):
11                 if self.friendsInfo[i][4] == CityList[j]:
12                     CityNum[j] += 1
13         for i in range(len(CityList)):
14             CityList[i] += '市'
15         # 生成 Map
16         map = Map("各市微信好友分佈", width=1000, height=800)
17         map.add("", CityList, CityNum, maptype="廣東", is_visualmap=True, visual_text_color='#000')
18         # 若文件名非空，則保存到該路徑下
19         if MapFile != "":
20             map.render(MapFile)
21             print(">>> Creat WeChat Map of Cities seccessfully!")

有了上述實現各個功能的方法，那麼就差一個調用各類方法的方法了。

(9) run方法

 1 def run(self):
 2     # 獲取微信好友信息
 3     self.getFriendsInfo()
 4     print(">>> Get WeChat friends' information successfully!")
 5     print(">>> Members:", self.allFriends_Num)
 6     # 保存微信好友信息
 7     self.saveFriendsInfoAsExcel(self.ExcelFile)
 8     # 分析微信好友信息
 9     self.quiteAnalyzeFriendsInfo()
10     # 使用微信好友的 city 產生詞雲圖
11     self.creatWordCloudOfCity(self.WCOfCityFile)
12     # 生成微信好友的 province 地圖
13     self.creatMapProvince(self.MapProvinceFile)
14     # 生成微信好友的 city 地圖
15     self.creatMapCity(self.MapCityFile)

對於文件路徑，在main函數中傳遞便可。【注】：上述代碼都在類中，在此處結束，下面爲main函數

1 if __name__ == "__main__":
2     ToExcelFile = "./WeChatAnalyze//FriendsInfo.xlsx"      # 微信好友信息的Excel表格保存路徑
3     ToPictureFile = "./WeChatAnalyze//CityWordCloud.png"   # 微信好友信息city詞雲圖保存路徑
4     ToMapFileProvince = "./WeChatAnalyze//WeChatProvinceMap.html" # 微信好友信息province地圖保存路徑
5     ToMapFileCity = "./WeChatAnalyze//WeChatCityMap.html"  # 微信好友信息city地圖保存路徑
6     # WeChatRobot對象實例化
7     robot = WeChatRobot(ToExcelFile, ToPictureFile, ToMapFileProvince, ToMapFileCity)

是否是以爲Main函數很簡短，哈哈，沒錯，就是這麼簡！

接下來看看實現的效果吧！

>>> 這個是終端顯示效果

>>> 這個是保存爲Excel表格的內容

 >>> 這個是微信好友各省的分佈

>>> 這個是微信好友各市的分佈

 

OK. 今天就分享到這啦！最後附上完整代碼！

 

參考文獻：

① 用python玩微信：http://www.javashuo.com/article/p-apsaupcm-cw.html

② pyecharts 中部分import 不到：https://blog.csdn.net/weixin_38617311/article/details/81146748

③ pyecharts 中地圖顯示不全：https://blog.csdn.net/xiamoyanyulrq/article/details/80025105

 

  1 # -*- coding: utf-8 -*-
  2 '''
  3 This is a program which can analyze datas of WeChat friends.
  4 @author: bpf
  5 '''
  6 
  7 #  導入模塊
  8 from wxpy import Bot
  9 import openpyxl
 10 from pyecharts import Map
 11 from wordcloud import WordCloud
 12 import matplotlib.pyplot as plt
 13 import jieba
 14 
 15 class WeChatRobot:
 16 
 17     '''====================== 1. 獲取微信好友信息 ======================'''
 18     def __init__(self, ToExcelFile="", ToCityFile="", ToMapProvinceFile="", ToMapCityFile=""):
 19         ''' 初始化機器人和其餘參數 '''
 20         # 初始化機器人，須要掃碼
 21         self.bot = Bot()
 22         # 獲取我全部的微信好友信息 - 存儲基礎信息(未處理)
 23         self.allFriends_Info = self.bot.friends()
 24         # 個人微信好友個數
 25         self.allFriends_Num = len(self.allFriends_Info)
 26         # 保存微信好友信息的表格文件路徑(.xlsx)
 27         self.ExcelFile = ToExcelFile
 28         # 保存城市詞雲圖的文件路徑(.png/.jpg)
 29         self.WCOfCityFile = ToCityFile
 30         # 保存省份地圖的文件路徑(.html)
 31         self.MapProvinceFile = ToMapProvinceFile
 32         # 其餘可用參數
 33         self.MapCityFile = ToMapCityFile
 34         # 自動調用run方法，使得在實例化對象後自動運行其餘函數
 35         self.run()
 36 
 37     '''====================== 2. 統計微信好友信息 ======================'''
 38     def getFriendsInfo(self):
 39         ''' 獲取微信好友的所有信息 '''
 40         # 存儲微信好友的信息(通過信息處理的)
 41         self.friendsInfo = []
 42         # 定義列標題
 43         self.infoTitle = ['NickName', 'RemarkName', 'Sex', 'Province', 'City']
 44         for aFriend in self.allFriends_Info:
 45             # 獲取暱稱
 46             NickName = aFriend.raw.get(self.infoTitle[0], None)
 47             # 獲取備註
 48             RemarkName = aFriend.raw.get(self.infoTitle[1], None)
 49             # 獲取性別
 50             Sex = {1:"男", 2:"女", 0:"其餘"}.get(aFriend.raw.get(self.infoTitle[2], None), None)
 51             # 獲取省份
 52             Province = aFriend.raw.get(self.infoTitle[3], None)
 53             # 獲取城市
 54             City = aFriend.raw.get(self.infoTitle[4], None)
 55             lisTmp = [NickName, RemarkName, Sex, Province, City]
 56             self.friendsInfo.append(lisTmp)
 57     
 58     '''====================== 3. 保存微信好友信息 ======================'''
 59     def saveFriendsInfoAsExcel(self, ExcelName):
 60         ''' 保存微信好友的信息到 Excel 表格中 '''
 61         # 生成openpyxl對象
 62         workbook = openpyxl.Workbook()
 63         # 激活表格
 64         sheet = workbook.active
 65         # 設置表格標題
 66         sheet.title = 'WeChatFriendsInfo'
 67         # 填充列標題到第一行
 68         for _ in range(len(self.infoTitle)):
 69             sheet.cell(row=1, column=_+1, value=self.infoTitle[_])
 70         # 填充微信好友信息，從第二行開始
 71         for i in range(self.allFriends_Num):
 72             for j in range(len(self.infoTitle)):
 73                 sheet.cell(row=i+2, column=j+1, value=str(self.friendsInfo[i][j]))
 74         # 若文件名非空，則保存到該路徑下
 75         if ExcelName != "":
 76             workbook.save(ExcelName)
 77             print(">>> Save WeChat friends' information successfully!")
 78 
 79     '''====================== 4. 分析微信好友信息 ======================'''
 80     def quiteAnalyzeFriendsInfo(self):
 81         ''' 分析數據，一步到位，直接了當 '''
 82         print(self.allFriends_Info.stats_text())
 83     
 84     '''====================== 5. 產生city詞雲圖 ======================'''
 85     def creatWordCloudOfCity(self, CityName):
 86         ''' 使用獲取的數據生成city詞雲圖 '''
 87         # 獲取全部的城市
 88         cityStr = ""
 89         for i in range(self.allFriends_Num):
 90             if self.friendsInfo[i][4] not in cityStr:
 91                 cityStr += " " + self.friendsInfo[i][4]
 92         #jieba庫精確模式分詞
 93         wordlist = jieba.lcut(cityStr)
 94         cityStr = ' '.join(wordlist)
 95         # 加載背景圖片
 96         #cloud_mask = np.array(Image.open(BackGroundFile))
 97         #設置詞雲圖屬性
 98         font = r'C:\Windows\Fonts\simfang.ttf' # 設置字體路徑
 99         wc = WordCloud(
100             background_color = 'black',     # 背景顏色
101             #mask = cloud_mask,             # 背景圖片
102             max_words = 100,                # 設置最大顯示的詞雲數
103             font_path = font,               # 設置字體形式（在本機系統中）
104             height = 300,                   # 圖片高度
105             width = 600,                    # 圖片寬度
106             max_font_size = 100,            # 字體最大值
107             random_state = 100,             # 配色方案的種類
108             )
109         # 生成詞雲圖
110         myword = wc.generate(cityStr)
111         #展現詞雲圖
112         plt.imshow(myword)
113         plt.axis('off')
114         plt.show()
115         # 若文件名非空，則保存到該路徑下
116         if CityName != "":
117             #保存詞雲圖
118             wc.to_file(CityName)
119             print(">>> Creat WeChat wordcloud of city successfully!")
120     
121     '''===================== 6. 產生province地圖 ====================='''
122     def creatMapProvince(self, MapFile):
123         ''' 使用獲取的數據生成province地圖 '''
124         # 獲取全部省份
125         provinceList, provinceNum = [], []
126         for i in range(self.allFriends_Num):
127             if self.friendsInfo[i][3] not in provinceList:
128                 provinceList.append(self.friendsInfo[i][3])
129                 provinceNum.append(0)
130         for i in range(self.allFriends_Num):
131             for j in range(len(provinceList)):
132                 if self.friendsInfo[i][3] == provinceList[j]:
133                     provinceNum[j] += 1
134         # 生成 Map
135         map = Map("各省微信好友分佈", width=1000, height=800)
136         map.add("", provinceList, provinceNum, maptype="china", is_visualmap=True, visual_text_color='#000')
137         # 若文件名非空，則保存到該路徑下
138         if MapFile != "":
139             #map.show_config()
140             map.render(MapFile)
141             print(">>> Creat WeChat Map of Provinces seccessfully!")
142     
143     '''===================== 7. 產生city地圖 ====================='''
144     def creatMapCity(self, MapFile):
145         ''' 使用獲取的數據生成city地圖 '''
146         # 獲取全部省份
147         CityList, CityNum = [], []
148         for i in range(self.allFriends_Num):
149             if self.friendsInfo[i][4] not in CityList:
150                 CityList.append(self.friendsInfo[i][4])
151                 CityNum.append(0)
152         for i in range(self.allFriends_Num):
153             for j in range(len(CityList)):
154                 if self.friendsInfo[i][4] == CityList[j]:
155                     CityNum[j] += 1
156         for i in range(len(CityList)):
157             CityList[i] += '市'
158         # 生成 Map
159         map = Map("各市微信好友分佈", width=1000, height=800)
160         map.add("", CityList, CityNum, maptype="廣東", is_visualmap=True, visual_text_color='#000')
161         # 若文件名非空，則保存到該路徑下
162         if MapFile != "":
163             map.render(MapFile)
164             print(">>> Creat WeChat Map of Cities seccessfully!")
165         
166     '''===================== 8. 自動執行函數 ====================='''
167     def run(self):
168         # 獲取微信好友信息
169         self.getFriendsInfo()
170         print(">>> Get WeChat friends' information successfully!")
171         print(">>> Members:", self.allFriends_Num)
172         # 保存微信好友信息
173         self.saveFriendsInfoAsExcel(self.ExcelFile)
174         # 分析微信好友信息
175         self.quiteAnalyzeFriendsInfo()
176         # 使用微信好友的 city 產生詞雲圖
177         self.creatWordCloudOfCity(self.WCOfCityFile)
178         # 生成微信好友的 province 地圖
179         self.creatMapProvince(self.MapProvinceFile)
180         # 生成微信好友的 city 地圖
181         self.creatMapCity(self.MapCityFile)
182         
183 if __name__ == "__main__":
184     ToExcelFile = "./WeChatAnalyze//FriendsInfo.xlsx"      # 微信好友信息的Excel表格保存路徑
185     ToPictureFile = "./WeChatAnalyze//CityWordCloud.png"   # 微信好友信息city詞雲圖保存路徑
186     ToMapFileProvince = "./WeChatAnalyze//WeChatProvinceMap.html" # 微信好友信息province地圖保存路徑
187     ToMapFileCity = "./WeChatAnalyze//WeChatCityMap.html"  # 微信好友信息city地圖保存路徑
188     # WeChatRobot對象實例化
189     robot = WeChatRobot(ToExcelFile, ToPictureFile, ToMapFileProvince, ToMapFileCity)