Python生成兔年祝福词云图

import numpy as np import wordcloud from PIL import Image import matplotlib.pyplot as plt from openpyxl import load_workbook import os #将存好的Excel词频表读取成字典 path= '\\兔年词云图\词频' #文件所在文件夹 files = [path+"\\"+i for i in os.listdir(path)] #获取文件夹下的文件名,并拼接完整路径 bk = np.array(Image.open('background.png')) # 定义词频背景图 for file in files: #将词频Excel文件读取为字典 wbc = load_workbook(file) wsc = wbc.active wordFreq = {} for i in range(2,wsc.max_row+1): word = wsc["A"+str(i)].value freq = wsc["B"+str(i)].value wordFreq[word] = freq #定义词云样式 wordc = wordcloud.WordCloud( font_path='C:/Windows/Fonts/simhei.ttf', # 设置字体 mode='RGBA', background_color='gold', mask= bk, # 设置背景图 colormap='hot', max_words=200, # 最多显示词数 max_font_size=150) # 字号最大值 #生成词云图 wordc.generate_from_frequencies(wordFreq) # 从字典生成词云 #保存图片到指定文件夹 wordc.to_file("1.png".format(file.split("\\")[1][:4])) #在notebook中显示词云图 plt.imshow(wordc) # 显示词云 plt.axis('off') # 关闭坐标轴 plt.show() # 显示图像