Python生成词云图wordcloud

1 环境

• jieba
• wordcloud
• PIL
• numpy
• matplotlib
• python3.5

2 例证

• Demo

import jieba
from wordcloud import WordCloud
from PIL import Image
import numpy as np
from matplotlib import pyplot as plt
# font = '/home/xdq/.local/share/fonts/SpaceMono-Regular.ttf'
font = '/usr/share/fonts/truetype/arphic/ukai.ttc'
def get_wordcloud(txt_dir, img_dir):
    with open(txt_dir, 'r',encoding='utf8') as f:
        content = f.read()
        print(content)
        cut = jieba.cut(content)
        print(cut)
        cut_content = ' '.join(cut)
        print(cut_content)
        # PIL打开图片
        img = Image.open(img_dir)
        # 获取图片矩阵信息
        img_array = np.array(img)
        # 设置目标图像属性:背景色,图层,字体,字体数量,尺寸
        wc = WordCloud(background_color='white', mask=img_array, font_path=font, max_words=1000, max_font_size=100, width=1800, height=1800)
        # 提取文字,生成云图内容
        img = wc.generate_from_text(cut_content)
        # 显示图片
        plt.imshow(img)
        # 保存图片
        wc.to_file('new.png')
# 文档文件        
txt_dir = "fullstack.txt"
# 原始图片(提取图片轮廓)
img_dir = "./images/7.jpeg"
get_wordcloud(txt_dir, img_dir)

• Result
  

图2.1 词云图

---