def genpic(): db = DB() table_name = "com_info" res = db.query(table_name) fig = plt.figure(figsize=(15, 15)) fig.set(alpha=0.5) df = pd.DataFrame(res) print(df) plt.subplot2grid((2, 2), (0, 0)) df.years.value_counts().plot(kind='bar') print(df.years.value_counts()) plt.subplot2grid((2, 2), (0, 1)) df.comscale.value_counts().plot(kind='bar') plt.subplot2grid((2, 2), (1, 0)) df.education.value_counts().plot(kind='bar') plt.subplot2grid((2, 2), (1, 1)) df.comdomain.value_counts().plot(kind='bar') plt.savefig('counts.jpg') plt.show()
if isfile(background_picture_filename): prefix = file.split('.')[0] bimg = imread(background_picture_filename) wordcloud = WordCloud(font_path=font_filename, background_color='white', mask=bimg, max_font_size=600, random_state=100) wordcloud = wordcloud.fit_words( dict(words_stat.head(4000).itertuples(index=False))) bimgColors = ImageColorGenerator(bimg) wordcloud.recolor(color_func=bimgColors) plt.imshow(wordcloud) plt.axis('off') plt.show() wordcloud.to_file("result.png") if __name__ == '__main__': db = DB() table_name = "com_info" res = db.query(table_name) content = "" for info in res: print(info['moredetail']) content = content + " " + info['moredetail'] main(content)