程序员的浪漫:词云照片 【wordcloud+jieba】
程序员文章站
2024-03-19 09:17:10
...
wordcloud:https://pypi.org/project/wordcloud/
jieba:https://pypi.org/project/jieba/
上述两个库分别是词云库和分词库(完美支持中文)
“““
导入各种库
”””
from PIL import Image
import jieba
import wordcloud
import matplotlib.pyplot as plt
import numpy as np
cloud_mask = plt.imread('1.jpg') #加载背景图片
with open('wordlist.txt','r') as f:
wordlist = f.read() #读取文本
print(wordlist)
WC = wordcloud.WordCloud(background_color = 'white',mask=cloud_mask,font_path='c:Windows/Fonts/STZHONGS.TTF',max_words=400,min_font_size=12,max_font_size=30,width=400,) #字体这里需要注意支持中文字体
WC.generate(wordlist)
WC.to_file('2.png')
plt.plot('2.png')
plt.show()
其中1.jpg为你想作为背景的图片,最好是背景与目标物体差别大的图片,个人建议可将原始图片经过灰度和去背景ps下,以突显任务轮廓
2.png为生成保存的图片
wordlist.txt为从网络上爬下来的形容女票的古诗句,重复度不是很高,会导致词分布很平均,但是可以通过改变字体参数
来改变想要突出的词语
该脚本已上传至github:https://github.com/guizi327832749/wordcloud
有什么说的不对的地方欢迎大家来批评指正~~