Python 几行代码生成词云还是随机颜色的呢

ghq322

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

#建立颜色数组，可更改颜色
color_list=['#FF7F00','#FF7F00','#ff00ff','#00ffff','#ffff00']
#调用
colormap=colors.ListedColormap(color_list)
# 分词
def trans_CN(text):
    # 接收分词的字符串
    word_list = jieba.cut(text)
    # 分词后在单独个体之间加上空格
    result = " ".join(word_list)
    return result


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



喜欢的拿去玩一下吧

ciker_li

感谢分享

chenxiang12159

很好 成功了龟龟

Bioyear

感谢分享

CWZ326237076

貌似numpy最新版1.19.4运行有问题，卸载掉默认安装的最新版numpy，指定安装1.19.3版本，好了。

hshcompass

谢谢分享。

hmhml

感谢分享，一会去试一下

Daoying

最后出来的图片无法保存，也没有本地路径

TAOBABY

楼主 我看这个字是随机的，请问如何设置成四个一组 四个一组的

niithua

兄弟，就不能换一个好看点的图形么？