You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
1021 B
38 lines
1021 B
2 months ago
|
import jieba
|
||
|
|
import wordcloud
|
||
|
|
import imageio
|
||
|
|
from wordcloud import ImageColorGenerator
|
||
|
|
|
||
|
|
#读取本地图片
|
||
|
|
img = imageio.imread('img.png')
|
||
|
|
#读取保存的弹幕数据
|
||
|
|
data = open('3.txt',encoding='utf-8')
|
||
|
|
text = data.read()
|
||
|
|
#print(text)
|
||
|
|
#分割词语
|
||
|
|
word = jieba.lcut(text)
|
||
|
|
print(word)
|
||
|
|
#转换为字符串
|
||
|
|
str = ' '.join(word)
|
||
|
|
print(str)
|
||
|
|
# 生成颜色映射
|
||
|
|
image_colors = ImageColorGenerator(img)
|
||
|
|
#绘制词云图
|
||
|
|
wc = wordcloud.WordCloud(
|
||
|
|
width=800,
|
||
|
|
height=800,
|
||
|
|
background_color='white',
|
||
|
|
mask=img,
|
||
|
|
contour_color='black',
|
||
|
|
contour_width=1,
|
||
|
|
stopwords={'哈','了','的','啊','你','是','我','吗','这','他','都','吧','这个','和','也','在','好','有','不','就','哈哈哈','不是','就是','说',
|
||
|
|
'真的','很','人','看','还','哈哈','哈哈哈哈','还是','没有','没','呢','什么','要','能','给'},
|
||
|
|
font_path='msyh.ttc'
|
||
|
|
|
||
|
|
|
||
|
|
)
|
||
|
|
wc.generate(str)
|
||
|
|
wc.recolor(color_func=image_colors)
|
||
|
|
wc.to_file("ai词云.png")
|
||
|
|
|