Page4

import numpy as N

file = '/home/peterh/Documents/DATA/dat/03987.dat'
dat = N.genfromtxt(file,names=True,dtype=None)

file = '/home/peterh/Documents/DATA/dat/gwlneudatum.dat'
gwl = N.genfromtxt(file,names=True,dtype=None)

id = N.where((dat['jahr']>=1961)&(dat['jahr']<=2020))[0]

ta = dat['ta'][id]
mo = dat['mo'][id]
ja = dat['jahr'][id]
tx = dat['tmax'][id]
pr = dat['nied'][id]

id = N.where((dat['jahr']>=1961)&(dat['jahr']<=2020))[0]

gw = gwl['gw'][id]
gw = N.array(gw,str)

gw[gw=='SWZ'] = 'SW'
gw[gw=='SWA'] = 'SW'

gw[gw=='SEZ'] = 'SE'
gw[gw=='SEA'] = 'SE'

gw[gw=='NWZ'] = 'NW'
gw[gw=='NWA'] = 'NW'

gw[gw=='TM']  = 'TRM'

gw[gw=='SZ']  = 'S'
gw[gw=='SA']  = 'S'

gw[gw=='NEZ'] = 'NE'
gw[gw=='NEA'] = 'NE'

gw[gw=='NZ']  = 'N'
gw[gw=='NA']  = 'N'

gw[gw=='WZ']  = 'W'
gw[gw=='WA']  = 'W'
gw[gw=='WW']  = 'W'
gw[gw=='WS']  = 'W'

gw[gw=='HNZ']  = 'HN'
gw[gw=='HNA']  = 'HN'

gw[gw=='HFZ']  = 'HF'
gw[gw=='HFA']  = 'HF'

gw[gw=='HNFZ'] = 'HNF'
gw[gw=='HNFA'] = 'HNF'

nd = len(gw)

f = open('includes/gwl.csv','w')
f.write('date,tmax,nied,gwl,year\n')

for d in range(nd):

    if((ja[d]>=1961)&(ja[d]<=1990)): dec = '1961-1990'
    if((ja[d]>=1991)&(ja[d]<=2000)): dec = '1991-2000'
    if((ja[d]>=2001)&(ja[d]<=2010)): dec = '2001-2010'
    if((ja[d]>=2011)&(ja[d]<=2020)): dec = '2011-2020'

    f.write('%04i-%02i-%02i,%.1f,%.1f,%s,%s\n'%(ja[d],mo[d],ta[d],tx[d],pr[d],gw[d],dec))

f.close()

f = open('./page4/includes/gwl.json','w')
f.write('[\n')

go = list(set(gw))

for g in go:

    id = N.where((pr>=20)&(gw==g))[0] 
    ii = N.where((gw==g))[0]

    f.write('\t{\n')
    f.write('\t\t"category": "%s",\n'%g)
    f.write('\t\t"amount": %.5f\n'%(100.*len(id)/len(ii)))
    f.write('\t},\n')

f.write(']\n')
f.close()