模仿大佬完成第一个Python项目（itchat学习）

学习itchat记录

今天看到CSDN上有大佬贴出使用itchat统计微信好友信息，正好想接触一下Python就实践了一下，特此记录：
我使用的环境是annaconda-Python2.7的版本

#引入itchat包
import itchat  
#登录自己的微信，会弹出二维码登录网页版微信
itchat.login()
#爬取自己好友信息，返回一个json文件
friends = itchat.get_friends(update=True)[0:]
#初始化计数器
male = female = other = 0
#由于friends[0]是自己的信息，所以从friends[1]开始统计好友数据
for i in friends[1:]:
    sex = i["Sex"]
    if sex == 1:
        male += 1
    elif sex ==2:
        female += 1
    else:
        other += 1
#计算朋友总数
total = len(friends[1:])
#打印出自己的好友性别比例
print("男性好友： %.2f%%" % (float(male)/total*100) + 
      "\n" + 
      "女性好友： %.2f%%" % (float(female)/total*100) + 
      "\n" +       
      "不明性别的好友: %.2f%%" % (float(other) / total * 100))

#以下是将好友信息保存到csv文件中**
#定义一个函数，用来爬取各个变量
def get_var(var):
    variable = []
    for i in friends:
        value = i[var]
        variable.append(value)
    return variable
#调用函数得到各变量，并把数据存到csv文件中，保存到桌面
NickName = get_var("NickName")
Sex = get_var('Sex')
Province = get_var('Province')
City = get_var('City')
Signature = get_var('Signature')
from pandas import DataFrame
data = {'NickName':NickName,'Sex':Sex,'Province':Province,
            'City':City,'Signature':Signature}
frame = DataFrame(data)
#由于中文乱码，微调了代码编码说明
frame.to_csv('data.csv',index=True,encoding='utf-8_sig')

#微信好友个性签名的自定义词云图
import re
siglist = []
#遍历好友签名信息
for i in friends:
    signature = i["Signature"].strip().replace("span","").replace("class","").replace("emoji","")
    rep = re.compile("1f\d+\w*|[<>/=]")
    signature = rep.sub("",signature)
    siglist.append(signature)
text = "".join(siglist)
#引入jieba包进行分词
import jieba
wordlist = jieba.cut(text, cut_all=True)
word_space_split = " ".join(wordlist)
#进入画图阶段
import matplotlib.pyplot as plt
from wordcloud import WordCloud, ImageColorGenerator
import numpy as np
import PIL.Image as Image
coloring = np.array(Image.open("C:/Users/Duke/Desktop/background.jpg")) //这是自定义的背景图片，大佬用了有轮廓的图像，可以让词云顺着轮廓分布，视觉效果满分
my_wordcloud = WordCloud(background_color="white",max_words=2000,
                         mask=coloring,max_font_size=60,random_state=42,scale=1,
                         font_path="C:/Users/Duke/Desktop/simkai.ttf").generate(word_space_split)
image_colors = ImageColorGenerator(coloring)
#图像显示部分
plt.imshow(my_wordcloud.recolor(color_func=image_colors))
plt.imshow(my_wordcloud)
plt.axis("off")
plt.show()

至此就结束了，第一次接触Python还是感觉挺好玩的