1、 功能
- 微信朋友性别分布
- 微信朋友省份分布
- 微信朋友北京分布
- 微信朋友个性签名词云
2、 源码：
import itchat
import pandas as pd
import numpy as np
from pyecharts import Pie, Map, Style, Page, Bar
import os
from PIL import Image
import jieba
from wordcloud import WordCloud, ImageColorGenerator
import matplotlib.pyplot as plt
def get_attr(friends, key):
return list(map(lambda user: user.get(key), friends))
def get_friends():
itchat.auto_login(hotReload=True)
friends = itchat.get_friends()
users = dict(province=get_attr(friends, "Province"),
city=get_attr(friends, "City"),
nickname=get_attr(friends, "NickName"),
sex=get_attr(friends, "Sex"),
signature=get_attr(friends, "Signature"),
remarkname=get_attr(friends, "RemarkName"))
return users
def sex_stats(users):
df = pd.DataFrame(users)
sex_arr = df.groupby(['sex'], as_index=True)['sex'].count()
data = dict(zip(list(sex_arr.index), list(sex_arr)))
data['不告诉你'] = data.pop(0)
data['帅哥'] = data.pop(1)
data['美女'] = data.pop(2)
return data.keys(), data.values()
def create_sex_charts(users, output_file):
page = Page()
style = Style(width=1100, height=600)
style_middle = Style(width=900, height=500)
data = sex_stats(users)
attr, value = data
chart = Pie('微信性别')
chart.add('', attr, value, center=[50, 50],
radius=[30, 70], is_label_show=True, legend_orient='horizontal', legend_pos='center',
legend_top='bottom', is_area_show=True)
page.add(chart)
page.render(output_file)
def prov_stats(users):
prv = pd.DataFrame(users)
prv_cnt = prv.groupby('province', as_index=True)['province'].count().sort_values()
attr = list(map(lambda x: x if x != '' else '未知', list(prv_cnt.index)))
return attr, list(prv_cnt)
def create_prov_charts(users, output_file):
data = prov_stats(users)
attr, value = data
page = Page()
style = Style(width=1100, height=600)
style_middle = Style(width=900, height=500)
chart = Map('中国地图', style.init_style)
chart.add('', attr, value, is_label_show=True, is_visualmap=True, visual_text_color='#000')
page.add(chart)
chart = Bar('柱状图', style_middle.init_style)
chart.add('', attr, value, is_stack=True, is_convert=True, label_pos='inside', is_legend_show=True,
is_label_show=True)
page.add(chart)
page.render(output_file)
def city_stats(users):
df = pd.DataFrame(users)
data = df.query('province == "北京"')
res = data.groupby('city', as_index=True)['city'].count().sort_values()
attr = list(map(lambda x: '%s区' % x if x != '' else '未知', list(res.index)))
return attr, list(res)
def create_city_charts(users, output_file):
data = city_stats(users)
attr, value = data
page = Page()
style = Style(width=1100, height=600)
style_middle = Style(width=900, height=500)
chart = Map('北京', style.init_style)
chart.add('', attr, value, maptype='北京', is_label_show=True, is_visualmap=True, visual_text_color='#000')
page.add(chart)
chart = Bar('柱状图', style_middle.init_style)
chart.add('', attr, value, is_stack=True, is_convert=True, label_pos='inside', is_label_show=True)
page.add(chart)
page.render(output_file)
def jieba_cut(users):
signature = users['signature']
words = ''.join(signature)
res_list = jieba.cut(words, cut_all=True)
return res_list
def create_wc(words_list):
res_png = os.path.join(os.getcwd(), 'wechat.png')
print (res_png)
words = ' '.join(words_list)
back_pic = np.array(Image.open(res_png))
stopwords = set()
stopwords = stopwords.union(set(['class','span','emoji','emoji','emoji1f388','emoji1f604']))
wc = WordCloud(background_color="white", margin=0,
font_path=os.getcwd()+'STXINGKA.TTF',
mask=back_pic,
max_font_size=70,
stopwords=stopwords
).generate(words)
plt.imshow(wc)
plt.axis('off')
plt.show()
if __name__ == '__main__':
users = get_friends()
sex_chart_file = os.getcwd() + 'sex_chart.html'
prov_chart_file = os.getcwd() + 'prov_chart.html'
city_chart_file = os.getcwd() + 'city_chart.html'
create_sex_charts(users, sex_chart_file)
print ('sex charts created')
create_prov_charts(users, prov_chart_file)
print ('prov charts created')
create_city_charts(users, city_chart_file)
print ('city charts created')
word_list = jieba_cut(users)
create_wc(word_list)
print ('word cloud created')
3、效果
