import numpy as np
import matplotlib.pyplot as plt

filename = "wb-data.dat"
data = np.genfromtxt(
    filename, skip_header=1, missing_values="..", delimiter=";"
)
years = np.linspace(1960, 2013, 54).astype(int)
cambodia_data, thailand_data, laos_data = data[:3], data[3:6], data[6:]


def plot_data(data, vaccine):
    plt.plot(years, data[0, 4:], lw=2, alpha=0.5, label="Cambodia")
    plt.plot(years, data[1, 4:], lw=2, alpha=0.5, label="Thailand")
    plt.plot(years, data[2, 4:], lw=2, alpha=0.5, label="Laos")
    plt.xlabel("Year")
    plt.ylabel(r"Immunization rate (\%)")
    plt.title("Immunization rates in children: ({})".format(vaccine))
    plt.legend(loc=4)
    plt.show()


plot_data(data[::3], "BCG")
plot_data(data[1::3], "Pol3")
plot_data(data[2::3], "Measles")