from wxpy import *   #導(dǎo)入模塊
 bot = Bot(cache_path=True) #初始化機(jī)器人，選擇掃碼登錄
 friend_all = bot.friends() #獲取微信好友信息

print(len(friend_all)) #好友的數(shù)量
 print(friend_all[0].raw) #輸出個(gè)人信息 

for a_friend in friend_all:
 NickName = a_friend.raw.get('NickName', None)
 #昵稱
 #Sex = a_friend.raw.get('Sex', None)
 Sex = {1: "男", 2: "女", 0: "其它"}.get(a_friend.raw.get('Sex', None), None)
 #性別（優(yōu)化）
 City = a_friend.raw.get('City', None)
 #城市
 Province = a_friend.raw.get('Province', None)
 #省份
 Signature = a_friend.raw.get('Signature', None)
 #個(gè)性簽名
 HeadImgUrl = a_friend.raw.get('HeadImgUrl', None)
 #頭像地址
 HeadImgFlag = a_friend.raw.get('HeadImgFlag', None)
 #小Flag
 list_0=[NickName, Sex, City, Province, Signature, HeadImgUrl, HeadImgFlag]
 #存為一維數(shù)組
 lis.append(list_0)
 #疊加數(shù)據(jù)

def list_excel(filename,lis):
 '''
 將列表寫入excel中，其中列表中的元素是列表.
 filename:保存的文件名（含路徑）
 lis：元素為列表的列表，如下：
 lis = [["名稱", "價(jià)格", "出版社", "語言"],
 ["暗時(shí)間", "32.4", "人民郵電出版社", "中文"],
 ["拆掉思維里的墻", "26.7", "機(jī)械工業(yè)出版社", "中文"]]
 '''
 import openpyxl
 wb = openpyxl.Workbook() #激活worksheet
 sheet = wb.active
 sheet.title = 'sheet1'  #創(chuàng)建一個(gè)表格
 file_name = filename +'.xlsx'
 for i in range(0, len(lis)):
  for j in range(0, len(lis[i])):
   sheet.cell(row=i+1, column=j+1, value=str(lis[i][j]))
   #每行每列的存入數(shù)據(jù)
 wb.save(file_name)
 print("寫入數(shù)據(jù)成功！")
list_excel('wechat',lis)

#導(dǎo)入模塊
from wordcloud import WordCloud
import matplotlib.pyplot as plt
import pandas as pd
from pandas import DataFrame
word_list= df['city'].fillna('0').tolist()
#將 dataframe 的列轉(zhuǎn)化為 list，其中的 nan 用“0”替換
new_text = ' '.join(word_list)
wordcloud = WordCloud(font_path='simhei.ttf', background_color="black").generate(new_text)
#設(shè)計(jì)圖背景顏色，字體
plt.imshow(wordcloud)
plt.axis("off")
plt.show()

#利用 pyechart 做詞云
import pandas as pd
#count = df.city.value_counts() #對 dataframe 進(jìn)行全頻率統(tǒng)計(jì)，排除了 nan
city_list = df['city'].fillna('NAN').tolist()#將 dataframe 的列轉(zhuǎn)化為 list，其中的 nan 用“NAN” 替換
count_city = pd.value_counts(city_list)#對 list 進(jìn)行全頻率統(tǒng)計(jì) 
from pyecharts.charts.wordcloud import WordCloud #設(shè)置對象
name = count_city.index.tolist()
value = count_city.tolist()
wordcloud = WordCloud(width=1300, height=620)
wordcloud.add("", name, value, word_size_range=[20, 100])
wordcloud.show_config()
wordcloud.render(r'D:\python\wechatcloud.html')

province_list = df['province'].fillna('NAN').tolist()
#將 dataframe 的列轉(zhuǎn)化為 list，其中的 nan 用 “NAN”替換
count_province = pd.value_counts(province_list)
#對 list 進(jìn)行全頻率統(tǒng)計(jì)

from pyecharts import Map
value =count_province.tolist()
attr =count_province.index.tolist()
map=Map("各省微信好友分布", width=1300, height=700)
map.add("", attr, value, maptype='china', is_visualmap=True,visual_text_color='#000',is_label_show = True)
#顯示地圖上的省份
map.show_config()
map.render(r'D:\python\wechatProMap.html')