python生成微信个性签名的词云图

需要用到的库：itchat，jieba，numpy，wordcloud

import itchat
import re
import jieba
import matplotlib.pyplot as plt
import PIL.Image as Image
import numpy as np
import os
from wordcloud import WordCloud,ImageColorGenerator
 
TList = []
def get_signature():
    friends = itchat.get_friends(update=True)[0:]
    for friend in friends:
        signature = friend["Signature"].strip()
        signature = re.sub("<span.*>", "", signature)
        signature = re.sub(" ", "", signature)
        signature = re.sub("emoji ", "", signature)
        TList.append(signature)
 
def cloud_pic():
    raw_signature_string = ''.join(TList)
    text = jieba.cut(raw_signature_string, cut_all=True)
    wl_space_split = ' '.join(text)
 
    d = os.path.dirname(os.path.abspath(__file__))
 
    alice_coloring = np.array(Image.open(os.path.join(d, "2.jpg"))) #原图
 
    my_wordcloud = WordCloud(background_color="white", #背景色
            max_words=2000,    #字数上限
            mask=alice_coloring, #形状
            max_font_size=50,#字体大小
            random_state=150, #随机数量
            font_path='C:/Windows/Fonts/simhei.ttf').generate(wl_space_split) #中文字体
    image_color = ImageColorGenerator(alice_coloring)
    plt.imshow(my_wordcloud.recolor(color_func=image_color))
    plt.imshow(my_wordcloud)
    plt.axis("off")
    plt.show()
 
def main():
    itchat.login()
    get_signature()
    cloud_pic()
    itchat.logout()
 
if __name__ == '__main__':
    main()
 
 

运行效果：

用到的背景图片2.jpg