• 之前看同學在群里發(fā)各人的消息云圖，然后很好奇就問了下怎么實現的自己也寫了一遍 2333

import re
import jieba
import matplotlib.pyplot as plt
from wordcloud import WordCloud
from scipy.misc import imread

• 這里是使用的包，用了 jieba 分詞和 wordcloud 云圖，matplotlib 用來繪圖，imread 用來讀取生成的云圖樣式

def seg_sentence(filename):
    message_regex = "(201\d-\d{2}-\d{2}) (\d{2}:\d{2}:\d{2}) (.*)\n(.*)(?=\n\n)"
    message_complile = re.compile(message_regex)
    with open(filename, encoding='utf-8') as f:
        message_data = message_complile.findall(f.read())
        words = ''
        for rows in message_data:
            if (rows[2] == str):
                seg_list = jieba.cut(rows[3], cut_all=False)
                seg_list = ' '.join(seg_list)
                if len(seg_list) > 1:
                    if seg_list != '\t':
                        words += seg_list
                        words += ' '
        return words

• 這里使用正則表達式用來匹配 QQ 群導出的 txt 格式消息記錄，然后調用 jieba 分詞對個人消息進行劃分，因為文本格式的消息記錄 發(fā)的表情和圖片都會變成 "[表情]"、"[圖片]"，一開始我是準備直接用 if 干掉的，后來把這兩條添加到了 stopwords 里

def draw_wordcloud(filename):
 
    words = seg_sentence(filename)
 
    font = 'C:\Windows\Fonts\simhei.ttf'
    color_mask = imread('test.jpg')
 
    cloud = WordCloud(font_path=font,
                      background_color='white',
                      mask=color_mask,
                      max_words=400,
                      prefer_horizontal=0.9,
                      stopwords=stopwordslist('stopwords.txt'),
                      # stopwords=None,
                      max_font_size=200)
    word_cloud = cloud.generate(words)
 
    # word_cloud.to_file("cloud.png")
    return word_cloud

• 繪圖的一些格式，這里的 test.jpg 就是我用電腦繪圖畫了一個云圖，用以生成自己想要的云圖格式，因為本地繪圖產生的圖片像素比較低，后來用 ps 把像素提高到 300dpi，這樣生成的圖片清晰度也高了很多

• stopwords 用的本地 stopwords，隨便下一個就好

def stopwordslist(filepath):
    stopwords = [line.strip() for line in open(filepath, 'r', encoding='utf-8').readlines()]
    return stopwords

• 最后就是讀取導出的 txt 格式消息記錄然后 plt 繪圖即可

plt.imshow(word_cloud)
plt.title(u'name', fontproperties='SimHei', fontsize='large')
plt.axis("off")
plt.show()