Create word cloud

master
pe79kpq4w 3 years ago
parent e9c6965d40
commit 3430fc72f1

@ -0,0 +1,19 @@
#生成词云
import wordcloud
import numpy as np
from PIL import Image
#生成词云的背景的图片数组
bgImage = ['CROWN.png', 'aSTAR.PNG', 'block.PNG']
def createcloud(str, filename, bgi):
print("----------------------生成词云中---------------------------")
image = Image.open(bgImage[bgi]) # 打开背景图
graph = np.array(image) # 读取背景图
w = wordcloud.WordCloud(font_path="C:\Windows\Fonts\msyhbd.ttc", background_color="black", mask=graph)
#str是已经处理好词频字符串
w.generate(str)
# 生成图片的名字,以文件名命名
png = "res" + filename.split('.')[0]
png += '.png'
w.to_file(png)
print("生成词云完成,请到文件夹中查看\n")
Loading…
Cancel
Save