python生成词云简单案例

环境：python3.6

第三方库：matplotlib、wordcloud、PIL

所需材料：一份英文love.txt文件(里面包含一篇英文文章)，一张心形图片love.jpg

代码如下：

from wordcloud import WordCloud
import PIL.Image as image
import numpy as np

with open("love.txt") as fp:
    text = fp.read()
    # print(text)
    mask = np.array(image.open("love.jpg"))
    wordcloud = WordCloud(
        mask=mask
    ).generate(text)
    image_produce = wordcloud.to_image()
    image_produce.show()

运行展示效果：

----------------------------分割线-------------------------------------------

 

以上使用于英文文本，中文文本流程如下

第三方库：matplotlib、wordcloud、PIL、jieba、numpy

所需材料：一份love.txt文件(里面包含一篇中文文章)，一张心形图片love.jpg

from wordcloud import WordCloud
import PIL.Image as image
import numpy as np
import jieba


# 分词
def trans_CN(text):
    # 接收分词的字符串
    word_list = jieba.cut(text)
    # 分词后在单独个体之间加上空格
    result = " ".join(word_list)
    return result


with open("love.txt",encoding="utf-8") as fp:
    text = fp.read()
    # print(text)
    # 将读取的中文文档进行分词
    text = trans_CN(text)
    mask = np.array(image.open("love.jpg"))
    wordcloud = WordCloud(
        # 添加遮罩层
        mask=mask,
        # 生成中文字的字体,必须要加,不然看不到中文
        font_path="C:\Windows\Fonts\STXINGKA.TTF"
    ).generate(text)
    image_produce = wordcloud.to_image()
    image_produce.show()

运行效果图如下：