wordcloud_quickstart_text_generation_matplotlib_display.py

1import os
2from os import path
3from wordcloud import WordCloud
4import matplotlib.pyplot as plt
5
6# get data directory (using wordcloud's own example text)
7d = path.dirname(__file__) if "__file__" in locals() else os.getcwd()
8
9# Read the whole text.
10text = open(path.join(d, 'constitution.txt')).read()
11
12# Generate a word cloud image
13wordcloud = WordCloud().generate(text)
14
15# Display the generated image:
16# the matplotlib way:
17plt.imshow(wordcloud, interpolation='bilinear')
18plt.axis("off")
19
20# lower max_font_size
21wordcloud = WordCloud(max_font_size=40).generate(text)
22plt.figure()
23plt.imshow(wordcloud, interpolation="bilinear")
24plt.axis("off")
25plt.show()