#!/usr/bin/env python
import numpy as np
import matplotlib.pyplot as plt
import re
from pprint import pprint as pp


# The following numbers were copied from http://www.cbi.ir/page/12660.aspx
# which is a PDF file from the official website of Central Bank of Iran.
# Also see: http://www.cbi.ir/showitem/12659.aspx (in Persian)

years = list(range(1386, 1393))
gini_index = [.4045, .3859, .3939, .3813, .3750, .3834, .3944]
inflation_rate = [18.4, 25.4, 10.8, 12.4, 21.5, 30.5, 34.7]

n = len(years)
xs = np.array(years)
ys = np.array(gini_index)
ys2 = np.array(inflation_rate)

plt.style.use('grayscale')
fig = plt.figure()
ax = fig.add_subplot(111)

axistexts = ax.set_xticklabels([str(x) for x in xs])
plt.setp(axistexts, rotation=-45)

plt.xlim(xs[0] - 1, xs[-1] + 1)
plt.xticks(xs)
plt.xlabel('Year')
plt.ylim(.37, .41)
plt.yticks(np.arange(.37, .41, .005))
plt.ylabel('Gini coefficient')


plt.plot(xs, ys, marker='o')

plt.grid(which='both')
plt.subplots_adjust(bottom=0.15)

plt.savefig('Gini_coefficient_Iran.svg')
plt.show()