1. 程式人生 > >python生成微信個性簽名的詞雲圖

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
在這裡插入圖片描述