observations.py 文件源码

def write_to_netcdf(self, ncfile):
        """ Write observation data to netcdf file """

        # Open ncfile and create coords
        dataset = Dataset(ncfile, 'w', format='NETCDF4_CLASSIC')
        zdim = dataset.createDimension('depth', self.z.size)
        tdim = dataset.createDimension('time', None)

        # Create time coordinate
        time = dataset.createVariable('time',np.float64,(tdim.name,))
        time.units = 'hours since 0001-01-01 00:00:00.0'
        time.calendar = 'gregorian'
        time[:] = date2num(self.dates, time.units, calendar=time.calendar)

        # Create depth coordinate 
        z = dataset.createVariable('depth',np.float64,(zdim.name,))
        z.units = 'm'
        z[:] = self.z

        # Create streamfunction variable
        sf = dataset.createVariable('stream_function_mar',np.float64,(tdim.name, zdim.name))
        sf.units = 'Sv'
        sf[:] = self.sf

        # Close file
        print 'SAVING: %s' % ncfile
        dataset.close()