# coding = utf-8

#引入微信登陆接口
from wxpy import *

#获取登录二维码
bot = Bot(cache_path = True)

#获取微信朋友的基本数据
friend_all = bot.friends()

#建立一个二维列表，存储基本好友信息
lis = [[‘NickName‘,‘Sex‘,‘City‘,‘Province‘,‘Signature‘,‘HeadImgUrl‘,‘HeadImgFlag‘]]
#把好有特征数据保存为列表
for a_friend in friend_all:
    NickName = a_friend.raw.get(‘NickName‘, None)
    Sex = {1:"男", 2:"女", 0:"其它"}.get(a_friend.raw.get(‘Sex‘, None), None)
    City = a_friend.raw.get(‘City‘, None)
    Province = a_friend.raw.get(‘Province‘, None)
    Signature = a_friend.raw.get(‘Signature‘, None)
    HeadImgUrl = a_friend.raw.get(‘HeadImgUrl‘, None)
    HeadImgFlag = a_friend.raw.get(‘HeadImgFlag‘, None)
    list_0 = [NickName, Sex, City, Province, Signature, HeadImgUrl, HeadImgFlag]
    lis.append(list_0)

#把列表转换为 xlsx 表格
def list_to_xlsx(filename, list):
   
    import openpyxl
    wb = openpyxl.Workbook()
    sheet = wb.active
    sheet.title = ‘Friends‘
    file_name = filename + ‘.xlsx‘
    for i in range(0, len(list)):
        for j in range(0, len(list[i])):
            sheet.cell(row = i+1, column = j+1, value = str(list[i][j]))
           
    wb.save(file_name)
    print("读写数据成功")

#把列表生成表格
list_to_xlsx(‘wechat_friend‘, lis)

   
from wordcloud import WordCloud
import matplotlib.pyplot as plt
import pandas as pd
from pandas import read_excel
import numpy as np

df = read_excel(‘wechat_friend.xlsx‘)

#使用 WordCloud 生成词云
word_list = df[‘City‘].fillna(‘0‘).tolist()
new_text = ‘ ‘.join(word_list)
wordcloud = WordCloud(font_path=‘msyh.ttc‘, background_color = ‘white‘).generate(new_text)
plt.imshow(wordcloud)
plt.axis("off")
plt.show()

#使用 pyecharts 生成词云
from pyecharts import WordCloud

city_list = df[‘City‘].fillna(‘‘).tolist()
count_city = pd.value_counts(city_list)
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:\wc.html‘)
wordcloud.render(‘wordcloud.html‘)

#将好友展示在地图上
from pyecharts import Map

province

_list = df[‘Province‘].fillna(‘‘).tolist()
count_province = pd.value_counts(province_list)
attr = count_province.index.tolist()
value1 = count_province.tolist()
map = Map("各省微信好友分布", width=1200,height=600)
map.add("",attr, value1, maptype=‘china‘,is_visualmap=True,visualmap_text_color=‘#000‘,is_label_show=True)
#map.show_config()
map.render(‘map.html‘)