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


# The following data were copied from
# http://alef.ir/vdceen8wpjh8wvi.b9bj.html?225846
text = """2.6 	33.3 	23.2 	41 	1384
2.6 	34.5 	23.4 	40.4 	1385
2.5 	35.5 	23.5 	39.8 	1386
2.4 	37.2 	22.8 	38 	1387
2.8 	37.3 	23.8 	38.9 	1388
3.2 	38.3 	23.8 	38.3 	1389
2.8 	39.8 	23.3 	36.9 	1390
2.9 	39.8 	24.1 	37.7 	1391
2.5 	39.5 	23.8 	37.6 	1392"""

unemployed = inactive = active = participation = years = ()
for line in text.split('\n'):
    data = line.split()
    unemployed += (float(data[0]),)
    inactive += (float(data[1]),)
    active += (float(data[2]),)
    participation += (float(data[3]),)
    years += (int(data[4]),)

ax = plt.subplot(211)

axistexts = ax.set_xticklabels([str(x) for x in years])
plt.setp(axistexts, rotation=-45, fontsize=10, clip_on=False)

plt.xlim(years[0], years[-1])
plt.xticks(years)
plt.ylim(
    min(unemployed + inactive + active) - 5,
    max(unemployed + inactive + active) + 5
)
plt.yticks(range(0, 41, 5))

plt.plot(years, unemployed, '#FF9900', marker='.', lw=1)
plt.plot(years, inactive, color='r', marker='.', lw=1)
plt.plot(years, active, color='b', marker='.', lw=1)

plt.grid(which='both')

ax = plt.subplot(212)

axistexts = ax.set_xticklabels([str(x) for x in years])
plt.setp(axistexts, rotation=-45, fontsize=10, clip_on=False)
plt.xticks(years)
plt.xlim(years[0], years[-1])
plt.xticks(years)
plt.plot(years, participation, marker='.', lw=1)
plt.grid(which='both')

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