自定义词云图片

ustbclearwang 2019-11-02

导包

from imageio import imread
import jieba
from wordcloud import WordCloud

自定义文本

str = ‘小明硕士毕业于中国科学院计算所,后在日本京都大学深造‘

从文件中导入文本

with open(‘文件名.txt‘, ‘r‘, encoding=‘utf8‘) as f:
    str = f.read()

剩下代码如下

# 利用jieba将字符串转化为单个词语的列表
ret = jieba.lcut(str)
# 根据 , 
s = ‘,‘.join(ret)

# 准备自定义图片
backgroud_Image = imread(‘头像.jpg‘)
print(‘图片准备成功‘)

# 生成词云对象
wordcloud = WordCloud(
    background_color=‘white‘,# 设置背景颜色
    mask=backgroud_Image,# 设置背景图片
    font_path=‘C:\Windows\Fonts\STZHONGS.TTF‘,  # 若是文本中有中文的话,这句代码必须添加,不然会出现方框,不出现汉字
    random_state=30,# 设置有多少种随机生成状态,即有多少种配色方案
).generate(s)

# 将生成对象书写在指定图片上
wordcloud.to_file(‘asdas.jpg‘)
print(‘开始加载图片‘)

相关推荐