Newer
Older
import numpy as np
from datetime import datetime, timedelta
from time import time
import os
os.getcwd()
start_date = datetime(2010, 1, 11)
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
kind = 'Density'
value = []
date_list = []
for i in range((end_date-start_date).days):
t0 = time()
c_date = start_date+timedelta(i)
filename = '/home/ess/cxt/work/data/WACCM/{}/{}/FWSD_2010.cam.h1.{}-00000.{}.2-5km.npz'.format(c_date.year, kind, c_date.strftime('%Y-%m-%d'), kind)
seconds = np.arange(0, 24*60*60, 60*30)
date_i = np.array([c_date+timedelta(0, int(seconds[i])) for i in range(seconds.size)])
if os.path.exists(filename) is True:
with np.load(filename) as data_load:
value.append(np.array(data_load['value'][1::2, -1].astype('float32')))
date_list.append(date_i)
t1 = time()
print('time usage:{} finish:{}'.format(t1-t0, filename))
else:
print('not found: {}'.format(filename))
value_all = np.concatenate(value, axis=0)
np.savez(kind, fields=value_all, date=np.array(date_list))
pass
kind = 'T'
value = []
date_list = []
for i in range((end_date-start_date).days):
t0 = time()
c_date = start_date+timedelta(i)
filename = '/home/ess/cxt/work/data/WACCM/{}/{}_height/FWSD_2010.cam.h1.{}-00000.{}.2-5km.npz'.format(c_date.year, kind, c_date.strftime('%Y-%m-%d'), kind)
seconds = np.arange(0, 24*60*60, 60*30)
date_i = np.array([c_date+timedelta(0, int(seconds[i])) for i in range(seconds.size)])
if os.path.exists(filename) is True:
with np.load(filename) as data_load:
value.append(np.array(data_load['value'][1::2, -1].astype('float32')))
date_list.append(date_i)
t1 = time()
print('time usage:{} finish:{}'.format(t1-t0, filename))
else:
print('not found: {}'.format(filename))
value_all = np.concatenate(value, axis=0)
np.savez(kind, fields=value_all, date=np.array(date_list))
kind = 'U'
value = []
date_list = []
for i in range((end_date-start_date).days):
t0 = time()
c_date = start_date+timedelta(i)
filename = '/home/ess/cxt/work/data/WACCM/{}/{}_height/FWSD_2010.cam.h1.{}-00000.{}.2-5km.npz'.format(c_date.year, kind, c_date.strftime('%Y-%m-%d'), kind)
seconds = np.arange(0, 24*60*60, 60*30)
date_i = np.array([c_date+timedelta(0, int(seconds[i])) for i in range(seconds.size)])
if os.path.exists(filename) is True:
with np.load(filename) as data_load:
value.append(np.array(data_load['value'][1::2, -1].astype('float32')))
date_list.append(date_i)
t1 = time()
print('time usage:{} finish:{}'.format(t1-t0, filename))
else:
print('not found: {}'.format(filename))
value_all = np.concatenate(value, axis=0)
np.savez(kind, fields=value_all, date=np.array(date_list))
kind = 'V'
value = []
date_list = []
for i in range((end_date-start_date).days):
t0 = time()
c_date = start_date+timedelta(i)
filename = '/home/ess/cxt/work/data/WACCM/{}/{}_height/FWSD_2010.cam.h1.{}-00000.{}.2-5km.npz'.format(c_date.year, kind, c_date.strftime('%Y-%m-%d'), kind)
seconds = np.arange(0, 24*60*60, 60*30)
date_i = np.array([c_date+timedelta(0, int(seconds[i])) for i in range(seconds.size)])
if os.path.exists(filename) is True:
with np.load(filename) as data_load:
value.append(np.array(data_load['value'][1::2, -1].astype('float32')))
date_list.append(date_i)
t1 = time()
print('time usage:{} finish:{}'.format(t1-t0, filename))
else:
print('not found: {}'.format(filename))
value_all = np.concatenate(value, axis=0)
np.savez(kind, fields=value_all, date=np.array(date_list))
kind = 'P'
value = []
date_list = []
for i in range((end_date-start_date).days):
t0 = time()
c_date = start_date+timedelta(i)
filename = '/home/ess/cxt/work/data/WACCM/{}/{}_height/FWSD_2010.cam.h1.{}-00000.lev.2-5km.npz'.format(c_date.year, kind, c_date.strftime('%Y-%m-%d'))
seconds = np.arange(0, 24*60*60, 60*30)
date_i = np.array([c_date+timedelta(0, int(seconds[i])) for i in range(seconds.size)])
if os.path.exists(filename) is True:
with np.load(filename) as data_load:
value.append(np.array(data_load['value'][1::2, -1].astype('float32')))
date_list.append(date_i)
t1 = time()
print('time usage:{} finish:{}'.format(t1-t0, filename))
else:
print('not found: {}'.format(filename))
value_all = np.concatenate(value, axis=0)
np.savez(kind, fields=value_all, date=np.array(date_list))