PISM_ARC/Getdata2.py at master · ac-johnson/PISM_ARC · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
Created on Mon Mar  5 10:30:01 2018

@author: andrew
"""

from IPython import get_ipython
get_ipython().magic('reset -sf')

#Get data

import numpy as np
import netCDF4
import sys
from glob import glob
from matplotlib import pyplot as plt

## Set up string to gather infiles ##

fileloc = '/home/andrewjohnson/runs/'
fns = 'ts_Ant_run'          #filename string
var1 = 'ice_volume_glacierized_grounded'
var2 = 'ice_area_glacierized_grounded'

runlist1=[4,5,6]
runlist2=[26,27,28]
runlist3=[7,8,9]
name='SIAe'


#runlist1=[10,11,12]
#runlist2=[29,30,31]
#runlist3=[13,14,15]
#name='SSAe'

iniset = netCDF4.Dataset(fileloc+fns+str(runlist1[0])+'.nc',mode='r')
times = (iniset['time'])[:]
tyr = times/31557600

#vol=np.zeros([])
ivol1=[]
ivol2=[]
ivol3=[]

iarea1=[]
iarea2=[]
iarea3=[]

for ii in runlist1:
    infile=fileloc+fns+str(ii)+'.nc'

    dset = netCDF4.Dataset(infile,mode='r')
    #print dset.variables.keys()

    rvol=(dset.variables[var1])[:]/1e9
    rarea = (dset.variables[var2])[:]/1e6

    ivol1.append(rvol)
    iarea1.append(rarea)

for ii in runlist2:
    infile=fileloc+fns+str(ii)+'.nc'

    dset = netCDF4.Dataset(infile,mode='r')
    #print dset.variables.keys()

    rvol=(dset.variables[var1])[:]/1e9
    rarea = (dset.variables[var2])[:]/1e6

    ivol2.append(rvol)
    iarea2.append(rarea)

for ii in runlist3:
    infile=fileloc+fns+str(ii)+'.nc'

    dset = netCDF4.Dataset(infile,mode='r')
    #print dset.variables.keys()

    rvol=(dset.variables[var1])[:]/1e9
    rarea = (dset.variables[var2])[:]/1e6

    ivol3.append(rvol)
    iarea3.append(rarea)

    #iarea = dset.variables['ice_area_glacierized']
    #farea=iarea[-1]

plt.figure(figsize=(13,6))
plt.plot(tyr,ivol1[0],label='SIAe=1',linewidth=3)
plt.plot(tyr,ivol1[1],label='SIAe=2',linewidth=3)
plt.plot(tyr,ivol1[2],label='SIAe=3',linewidth=3)
plt.plot(tyr,ivol2[0],'--',color='blue',linewidth=2)
plt.plot(tyr,ivol2[1],'--',color='green',linewidth=2)
plt.plot(tyr,ivol2[2],'--',color='red',linewidth=2)
plt.plot(tyr,ivol3[0],'-.',color='blue',linewidth=3)
plt.plot(tyr,ivol3[1],'-.',color='green',linewidth=3)
plt.plot(tyr,ivol3[2],'-.',color='red',linewidth=3)
#plt.plot(tyr,ivol2[1],label='SIAe=2',linewidth=3)
#plt.plot(tyr,ivol2[2],label='SIAe=3',linewidth=3)
plt.legend(fontsize=20,loc=2)
plt.tick_params(axis='both', labelsize=25)
plt.title('Grounded Ice Volume',fontsize=35)
plt.xlabel('time (years)',fontsize=25)
plt.ylabel('volume (1e7 km^3)',fontsize=25)
plt.savefig('figures/volume_'+name+'.png')

plt.figure(figsize=(13,6))
plt.plot(tyr,iarea1[0],label='SIAe=1',linewidth=3)
plt.plot(tyr,iarea1[1],label='SIAe=2',linewidth=3)
plt.plot(tyr,iarea1[2],label='SIAe=3',linewidth=3)
plt.plot(tyr,iarea2[0],'--',color='blue',linewidth=2)
plt.plot(tyr,iarea2[1],'--',color='green',linewidth=2)
plt.plot(tyr,iarea2[2],'--',color='red',linewidth=2)
plt.plot(tyr,iarea3[0],'-.',color='blue',linewidth=3)
plt.plot(tyr,iarea3[1],'-.',color='green',linewidth=3)
plt.plot(tyr,iarea3[2],'-.',color='red',linewidth=3)
plt.legend(fontsize=20,loc=2)
plt.title('Grounded Ice Area',fontsize=35)
plt.xlabel('time (years)',fontsize=25)
plt.ylabel('Area (1e7 km^2)',fontsize=25)
plt.tick_params(axis='both', labelsize=25)
plt.savefig('figures/area_'+name+'.png')