Wordcloud词云绘制关键词

import jieba  # 分词
from matplotlib import pyplot as plt  # 绘图数据可视化
from wordcloud import WordCloud  # 词云
from PIL import Image  # 图片处理
import numpy as np  # 矩阵运算

text = open("DS.txt", encoding='utf-8').read()
# 分词
cut = jieba.cut(text)  # 返回一个对象
string = ' '.join(cut)
print(string)
# 将背景图片进行处理
img = Image.open("pure.png")  # 打开遮罩图片
img_array = np.array(img)  # 将图片转化为数组
# 设置词云相关参数
wc = WordCloud(
    background_color='white',
    mask=img_array,
    font_path="msyh.ttc",
)
wc.generate_from_text(string)
# 绘制图片
fig = plt.figure(1)
plt.imshow(wc)  # 将图片以wc方式显示出来
plt.axis('off')  # 是否显示坐标轴
plt.show()
wc.to_file('dsm.png')