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)