快捷導(dǎo)航

Python實現(xiàn)的微信好友數(shù)據(jù)分析功能示例

更新時間：2018年06月21日 10:30:20 作者：騎著炮彈進(jìn)城

這篇文章主要介紹了Python實現(xiàn)的微信好友數(shù)據(jù)分析功能,結(jié)合實例形式分析了Python使用itchat、pandas、pyecharts等模塊針對微信好友數(shù)據(jù)進(jìn)行統(tǒng)計與計算相關(guān)操作技巧,需要的朋友可以參考下

本文實例講述了Python實現(xiàn)的微信好友數(shù)據(jù)分析功能。分享給大家供大家參考，具體如下：

這里主要利用python對個人微信好友進(jìn)行分析并把結(jié)果輸出到一個html文檔當(dāng)中，主要用到的python包為itchat，pandas，pyecharts等

1、安裝itchat 微信的python sdk，用來獲取個人好友關(guān)系。獲取的代碼如下：

import itchat
import pandas as pd
from pyecharts import Geo, Bar
itchat.login()
friends = itchat.get_friends(update=True)[0:]
def User2dict(User):
  User_dict = {}
  User_dict["NickName"] = User["NickName"] if User["NickName"] else "NaN"
  User_dict["City"] = User["City"] if User["City"] else "NaN"
  User_dict["Sex"] = User["Sex"] if User["Sex"] else 0
  User_dict["Signature"] = User["Signature"] if User["Signature"] else "NaN"
  User_dict["Province"] = User["Province"] if User["Province"] else "NaN"
  return User_dict
friends_list = [User2dict(i) for i in friends]
data = pd.DataFrame(friends_list)
data.to_csv('wechat_data.csv', index=True)

2、對獲取到的數(shù)據(jù)進(jìn)行分析。

主要分析了男女比例，以及好友所在城市分布，并且在地圖上面展示了微信好友的分布情況。另外其他的數(shù)據(jù)讀者可以自己去分析，這里只是提供一個引導(dǎo)而已。

import pandas as pd
from pyecharts import Geo, Bar
def Cal_mVw(data):
  result = {}
  for i in data:
    if i == 1:
      result["man"] = result.get("man", 0) + 1
    elif i == 2:
      result["woman"] = result.get("woman", 0) + 1
    else:
      result["unknown"] = result.get("nunknown", 0) + 1
  return result
def count_city(data):
  result = {}
  for i in data:
    if data is not "NaN" or data is not "nan":
      result[i] = result.get(i, 0) + 1
  return result
data1 = pd.read_csv('wechat_data.csv', encoding='GBK')
manVSwoman=Cal_mVw(data1["Sex"])
#print(manVSwoman)
bar = Bar("個人微信好友男女比例")
bar.add("男女人數(shù)", ["男", "女", "不詳"], [139, 75, 1])
bar.render()
city=count_city(data1["City"])
geo = Geo("微信好友分布", "", title_color="#fff", title_pos="center",
width=1200, height=600, background_color='#404a59')
#attr, value = geo.cast(city)
geo.add("", city.keys(), city.values(), visual_range=[0, 30], visual_text_color="#fff", symbol_size=15, is_visualmap=True)
geo.show_config()
geo.render()

男女比例畫出來的圖如下所示