PythonCode/fileforloop.py at master · PaulSpence/PythonCode · 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
"""
Write purpose
"""

import pdb
import numpy as np
import netCDF4 as nc4

import glob
import os
import logging


basepath = '../gfdl_nyf_1080/output*'

def slurp(path, vname):
    """ Read a variable while closing the file immediately
    """
    ifile = nc4.Dataset(path)
    var = ifile.variables[vname][:]
    ifile.close()
    return var

geolat_c = slurp('../gfdl_nyf_1080/output101/ocean_grid.nc','geolat_c')
geolon_c = slurp('../gfdl_nyf_1080/output101/ocean_grid.nc','geolat_c')
tx_trans = slurp('../gfdl_nyf_1080/output101/ocean.nc','tx_trans')
sw_ocean = slurp('../gfdl_nyf_1080/output101/ocean.nc','sw_ocean')

runs = sorted(glob.glob(basepath))
pdb.set_trace()


for path in runs[20:]: #will process runs 20 forward

    tx_trans = slurp(os.path.join(path,'ocean.nc'), 'tx_trans')