# Stacked Bar Chart with Matplotlib
# Data from https://en.wikipedia.org/wiki/Wind_power_in_the_Netherlands

import numpy as np
import matplotlib.pyplot as plt

# label of y-axis and title
plt.title('Wind power installed capacity in the Netherlands in MW')
plt.ylabel('MW')

#Format= ['show y/n', year , land-based, sea-based]
data=[
['x',2000, 447 , 0  ] ,   
[' ',2001, 485 , 0  ] ,
[' ',2002, 672 , 0  ] ,
[' ',2003, 905 , 0  ] ,
[' ',2004, 1075, 0  ] ,
['x',2005, 1224, 0  ] ,
[' ',2006, 1453, 108 ] ,
[' ',2007, 1641, 108 ] ,
[' ',2008, 1921, 228 ] ,
[' ',2009, 1994, 228 ] ,    
['x',2010, 2009, 228 ] ,
[' ',2011, 2088, 228 ] ,
[' ',2012, 2205, 228 ] ,
[' ',2013, 2485, 228 ] ,
[' ',2014, 2637, 228 ] ,
['x',2015, 3034, 357 ] ,
[' ',2016, 3300, 957 ] ,
[' ',2017, 3245, 957 ] ,
[' ',2018, 3436, 957 ] ,
[' ',2019, 3527, 957 ] ,
['x',2020, 4188, 2460] , 
[' ',2021, 5214, 2460] ,
[' ',2022, 6185, 2570] , 
[' ',2023, 6771, 3978] 
]
# please update in future.

show_yesno = np.array([row[0] for row in data])
all_years  = np.array([row[1] for row in data])
bar_labels = []

# show axis-labels only the years marked with 'x'
for i, yesNo in enumerate(show_yesno):
    if (yesNo == 'x'):
        bar_labels.append(all_years[i])
    else:
        bar_labels.append("")
        

#bar_labels = np.array([row[1] for row in data])
data1      = np.array([row[2] for row in data])
data2      = np.array([row[3] for row in data])

plt.xticks(range(len(data1)), bar_labels  )

plt.bar(range(len(data1)), data1 ,  label='Land-based capacity (MW)')
plt.bar(range(len(data2)), data2, bottom=data1,  label='Sea-based capacity (MW)')
plt.legend()

plt.savefig('Wind_power_installed_capacity_in_NL_MW.svg', format='svg')
plt.show()