在工作里通常会碰到做词云图,以便于更好的展现主题.
例如这样:
我们会有关键词(此案例对应的为清洗版)以及频数。通过他我们生成词云图。
我们看一下代码:
#这里是我们需要引用到的库
import pandas as pd
import matplotlib.pyplot as plt
from wordcloud import WordCloud
#这里是我们拿到代码跑的时候自己需要更改的地方。
#1.这行代码中第一个引号是指的你需要处理文件夹的位置,通常将文件拖到与代码文件在同一路径下即可。
#2.第二个引号‘情绪词 消极’是代表上图中下面各个分表(sheet),你想要生词那张表的词云图就引号中的替换成那张表。
# 获得数据
data=pd.read_excel(r'C:\Users\Yinfinity\PycharmProjects\Test\基金超话分类 2.0.xlsx',sheet_name='情绪词 消极')
# 求频次
data_gr= pd.DataFrame(data)
# print(data_gr)
#这里也是可能需要改的地方。
#这里的清洗版和频数对应的是第一行的名字,如第二张图是关键词,就将下面代码中的清洗版换成关机词即可。
# 转化为字典形式
dic=dict(zip(data_gr['清洗版'],data_gr['频数']))
# print(dic)
#fit_word函数,接受字典类型,其他类型会报错
wordcloud = WordCloud(font_path='simhei.ttf',background_color="white",width =4000,height= 2000,margin= 10 ).fit_words(dic)
print(wordcloud)
plt.imshow(wordcloud)
# 显示
plt.show()
注:
对于文件夹位置拖到pycharm后,右键点击文件,点击copy path,再点击第二张图的位置即可。