File size: 809 Bytes
bd3493c | 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 | """
Data is stored in memmaps for access speed. Here we give the shapes required to load these files
"""
CLIMATOLOGY_SHAPE = (4, 366, 24, 240, 121)
ICOADS_Y_SHAPE = (33601, 5, 12000)
ICOADS_X_SHAPE = (33601, 2, 12000)
IGRA_Y_SHAPE = (33604, 24, 1375)
IGRA_X_SHAPE = (1375, 2)
AMSUA_Y_SHAPE = (21916, 180, 360, 13)
AMSUB_Y_SHAPE = (21916, 360, 181, 12)
ASCAT_Y_SHAPE = (21913, 360, 181, 17)
HIRS_Y_SHAPE = (21913, 360, 181, 26)
GRIDSAT_Y_SHAPE = (48211, 2, 514, 200)
IASI_Y_SHAPE = (23373, 360, 181, 52)
def get_hadisd_shape(mode):
"""
Return the shape of the HadISD array depending on variable
"""
if mode != "train":
dim_1 = 415
else:
var_dict = {"tas": 8719, "tds": 8617, "psl": 8016, "u": 8721, "v": 8721}
dim_1 = var_dict[var]
return (106652, dim_1)
|