個(gè)人博客

http://www.milovetingting.cn

背景圖

mask.jpg

效果圖

sample.jpg

代碼

from os import path
import jieba
from wordcloud import WordCloud
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt


def handle_data():
    """
    處理文本數(shù)據(jù)
    :return:
    """
    # 讀取數(shù)據(jù)
    with open("data.txt", "r") as f:
        txt = f.read()

    # 去除無效數(shù)據(jù)
    re_move = ["，", "。", " ", '\n', '\xa0']
    for i in re_move:
        txt = txt.replace(i, " ")

    # 使用精確分詞模式
    word = jieba.lcut(txt)

    # 保存數(shù)據(jù)
    with open("data_handled.txt", 'w') as file:
        for i in word:
            file.write(str(i) + ' ')


def generate_image():
    """
    生成圖片
    :return:
    """
    # 讀取數(shù)據(jù)
    with open("data_handled.txt", "r") as file:
        txt = file.read()

    # 圖片路徑
    d = path.dirname(__file__)

    # 生成mask
    mask = np.array(Image.open(path.join(d, "mask.jpg")))

    # 生成word
    word = WordCloud(
        background_color="white",
        width=800,
        height=800,
        mask=mask,
        # 字體路徑,WordCloud默認(rèn)不支持中文,這里的SimHei.ttf需要下載放到系統(tǒng)字體庫(kù)目錄下
        font_path='SimHei.ttf'
    ).generate(txt)

    # 保存圖片
    word.to_file('world_cloud.png')

    # 使用plt庫(kù)顯示圖片
    plt.imshow(word)

    plt.axis("off")

    plt.show()


if __name__ == '__main__':
    handle_data()
    generate_image()

參考

https://blog.csdn.net/loveyouandc/article/details/88193641

https://www.cnblogs.com/djdjdj123/p/12153603.html