Python学习笔记--获取好友信息

待我称王封你为后i 2022-05-22 06:37 183阅读 0赞

参考链接：[https://zhuanlan.zhihu.com/p/36418880][https_zhuanlan.zhihu.com_p_36418880]  
一、代码实现：

# -*- coding:utf-8 -*-
    # 微信撤回消息
    # python2
    # 参考链接：https://zhuanlan.zhihu.com/p/36418880
    
    import itchat
    itchat.login()
    
    # 爬取自己好友相关信息，返回一个json文件
    friends = itchat.get_friends(update=True)[0:]
    #print friends
    
    # 1、获得微信好友性别比例
    male = female = other = 0
    for i in friends[1:]: # friengs[0]是自己的信息
        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)
    
    # 2、获得微信好友城市分布
    # 首先定义一个函数爬去各个变量
    
    def get_var(var):
        variable = []
        for i in friends:
            value = i[var]
            if type(value) != int:
                value = value.encode('utf-8')
            variable.append(value)
        return variable
    Nickname = get_var('NickName')
    #print Nickname
    Sex = get_var('Sex')
    Province = get_var('Province')
    City = get_var('City')
    RemarkName = get_var('RemarkName') # 备注名
    Signature = get_var('Signature')
    
    from pandas import DataFrame
    data = {
       "NickName":Nickname,"Sex":Sex,"Province":Province,"City":City,"Signature":Signature,"RemarkName":RemarkName}
    frame = DataFrame(data)
    frame.to_csv('data.csv')
    
    # 3、微信好友签名自定义云图
    # 首先将签名表示出来，并去除各种表情什么的，再用正则表达式去除《>=等符号
    import re
    siglist = []
    for i in friends:
        signature = i['Signature'].encode('utf-8').strip().replace('span','').replace('class','').replace('emoji',"")
        print signature
        #siglist.append(signature)
        rep = re.compile("1f\d+\w*|[<>/=]") # 正则表达式？？？没懂。。。
        signature = rep.sub('',signature)
        siglist.append(signature)
        print siglist
    text = "".join(siglist)
    print text.replace(" ","").replace("\n","")
    
    # 结巴分词包,对字符串进行分词
    # 它是Python最好的中文分词组件；
    import jieba
    wordlist = jieba.cut(text,cut_all=True) # 全模式;False为精确模式；无为默认模式；cut_search_for()为搜索引擎模式
    word_space_split = " ".join(wordlist)
    print word_space_split
    
    # 画图
    import matplotlib.pyplot as plt
    from wordcloud import WordCloud,ImageColorGenerator
    import numpy as np
    import PIL.Image as Image
    
    
    coloring = np.array(Image.open(r"C:\Users\ZHYU\PycharmProjects\wechat\1.png"))
    my_wordcloud = WordCloud(background_color='white',max_words=2000,mask=coloring,max_font_size=60,random_state=42,\
                             scale=2,font_path=r"C:\Users\ZHYU\Downloads\NotoSans-hinted\NotoSans-Black.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()
    
    itchat.logout()

运行结果：  
![这里写图片描述][70]

二、遇到的问题  
1、编码问题：由于用search\_friends()得到的是所有好友信息的json数据，所以得到的昵称、签名等都是u’\\u789’这种unicode形式，需要将其转为汉字表示时，用encode(“utf-8”)即可。

2、安装PIL遇到的问题，直接用pip安装后，运行报错：

ImportError: The _imagingft C module is not installed

经过百度找到了解决方案，在此链接上下载与你的系统对应的pillow版本  
[https://www.lfd.uci.edu/~gohlke/pythonlibs/\#pillow][https_www.lfd.uci.edu_gohlke_pythonlibs_pillow]即可成功运行！

3、最后的云图出现了乱码……此问题还未解决……下午再看

[https_zhuanlan.zhihu.com_p_36418880]: https://zhuanlan.zhihu.com/p/36418880
[70]: /images/20220522/0bb531bf4ca84f1a849f400ed5a119c6.png
[https_www.lfd.uci.edu_gohlke_pythonlibs_pillow]: https://www.lfd.uci.edu/~gohlke/pythonlibs/#pillow