Demonstrate vega Data Sets

#Iowa electricity consumtion

from vega_datasets import data

from matplotlib import pyplot as plt

import pandas as pd

l=data.iowa_electricity()

print(l)

#print(l['source'])

datelist=[]

nuclear=[]

fossil=[]

renewable=[]

for x in range(0,len(l)):

s=str(l['year'][x])

s=s[:4]

yr=int(s)

if yr not in datelist:

datelist.append(yr)

val=int(l['net_generation'][x])

#print(l['source'][x])

if l['source'][x] == 'Fossil Fuels':

#print(s)

fossil.append(val)

elif l['source'][x] == 'Nuclear Energy':

nuclear.append(val)

elif l['source'][x] == 'Renewables':

renewable.append(val)

plt.xlabel('Year')

plt.ylabel('Power Cunsumtion in Kw')

plt.plot(datelist,fossil,label='Fossil Fuels')

plt.plot(datelist,nuclear,label='Nuclear Energy')

plt.plot(datelist,renewable,label='Renewable Energy')

plt.legend( loc='best', borderaxespad=0.)

plt.show()