Matplot incluindo annotates

matplot-incuindo-annotates.py

import pandas as pl
import matplotlib.pyplot as plt

"""
Incluindo annotates
"""
dataframe = pl.read_csv('annual-real-gnp-us-1909-to-1970.csv')

plt.figure(figsize=(8, 6))
plt.plot(dataframe['Year'], dataframe['GNP'])
plt.xlabel('Year')
plt.ylabel('GNP')
plt.title('Annual Real USA GNP 1909 to 1970')

# matplotlib.pyplot.annotate(*args, **kwargs)
# s:str - The text of the annotation
# xy:iterable - Length 2 sequence specifying the (x,y) point to annotate
# xytext : iterable, optional - Length 2 sequence specifying the (x,y)
#          to place the text at. If None, defaults to xy.
# arrowprops: dict, optional
plt.annotate('Fim da Segunda Gerra Mundial', xy=(1945, 355.2),
             arrowprops=dict(arrowstyle='->'), xytext=(1920, 300))
plt.annotate('Crash da Bolsa', xy=(1929, 203.6),
             arrowprops=dict(arrowstyle='->'), xytext=(1909, 200))

plt.savefig('gnp-us-1909-to-1970-annotate.png')

plt.close()