Python loadmat示例

示例#1

文件： load_spec_samples.py 项目： tmiesse/PolyADCIRC

import numpy as np
import scipy.io as sio
from scipy.interpolate import griddata

# Specify run parameter folders
base_dir = '/h1/lgraham/workspace'
grid_dir = base_dir + '/ADCIRC_landuse/Inlet_b2/inputs/for_walls'
save_dir = base_dir + '/ADCIRC_landuse/Inlet_b2/runs/for_wall_q2'
basis_dir = base_dir + '/ADCIRC_landuse/Inlet_b2/gap/beach_walls_2lands'

# setup and save to shelf
# set up saving
save_file = 'py_save_file'

# Load and fix water heights
main_run, domain, mann_pts, wall_pts, points = rmw.loadmat(
    save_file, base_dir, grid_dir, save_dir, basis_dir)

mdict = dict()
mdict['time_obs'] = main_run.time_obs
mdict['ts_data'] = main_run.ts_data
mdict['nts_data'] = main_run.nts_data
mdict['mann_pts'] = mann_pts

# Interpolate station data for non-time series data
# Change xi to add more or different stations
xi = np.array([[s.x, s.y] for s in domain.stations['fort61']])
points = np.column_stack((domain.array_x(), domain.array_y()))
orig = main_run.nts_data.copy()
for key, values in orig.iteritems():
    if key[-1] == str(3):
        main_run.nts_data[key[:-1] + '1'] = griddata(points, values, xi)

示例#2

文件： concatenate_many.py 项目： UT-CHG/PolyADCIRC

#! /usr/bin/env python
# import necessary modules
import polyadcirc.run_framework.random_wall as rmw

base_dir = '/h1/lgraham/workspace'
grid_dir = base_dir + '/ADCIRC_landuse/Inlet/inputs/poly_walls'
save_dir = base_dir + '/ADCIRC_landuse/Inlet/runs/poly_wall'
basis_dir = base_dir +'/ADCIRC_landuse/Inlet/landuse_basis/gap/beach_walls_2lands'

# set up saving
save_file = 'py_save_file'

main_run, domain, mann_pts, wall_pts, points = rmw.loadmat(save_file+'0',
        base_dir, grid_dir, save_dir+'_0', basis_dir)


for i in xrange(1, 7):
    save_file2 = save_file+str(i)
    save_dir2 = save_dir+'_'+str(i)
    other_run, domain, mann_pts2, wall_pts2, points2 = rmw.loadmat(save_file2, 
            base_dir, grid_dir, save_dir2, basis_dir)
    run, points = main_run.concatenate(other_run, points, points2)

mdat = dict()
mdat['points'] = points

main_run.update_mdict(mdat)
main_run.save(mdat, 'poly7_file')

示例#3

文件： concatenate_pair.py 项目： UT-CHG/PolyADCIRC

#! /usr/bin/env python
# import necessary modules
import polyadcirc.run_framework.random_wall as rmw
import numpy as np

base_dir = '/h1/lgraham/workspace'
grid_dir = base_dir + '/ADCIRC_landuse/Inlet/inputs/poly_walls'
save_dir1 = base_dir + '/ADCIRC_landuse/Inlet/runs/poly_wall1'
save_dir2 = base_dir + '/ADCIRC_landuse/Inlet/runs/poly_wall2'
basis_dir = base_dir +'/ADCIRC_landuse/Inlet/landuse_basis/gap/beach_walls_2lands'

# setup and save to shelf
# set up saving
save_file = 'py_save_file'

main_run, domain, mann_pts1, wall_pts1, points1 = rmw.loadmat(save_file,
        base_dir, grid_dir, save_dir1, basis_dir)
other_run, domain, mann_pts2, wall_pts2, points2 = rmw.loadmat(save_file,
        base_dir, grid_dir, save_dir2, basis_dir)

cated = main_run.concatenate(other_run, points1, points2)

mdat = dict()
mdat['points'] = cated[1]
mdat['mann_pts'] = np.concatenate((mann_pts1, mann_pts2), axis=points1.ndim-1)
mdat['wall_pts'] = np.concatenate((wall_pts1, wall_pts2), axis=points1.ndim-1)

main_run.update_mdict(mdat)
main_run.save(mdat, 'cat_file')

示例#4

文件： load_random_wall.py 项目： UT-CHG/PolyADCIRC

#! /usr/bin/env python
# import necessary modules
import polyadcirc.run_framework.random_wall as rmw
import polyadcirc.pyADCIRC.plotADCIRC as pa

base_dir = '/h1/lgraham/workspace'
grid_dir = base_dir + '/ADCIRC_landuse/Inlet/inputs/poly_walls'
save_dir = base_dir + '/ADCIRC_landuse/Inlet/runs/few_walls'
basis_dir = base_dir +'/ADCIRC_landuse/Inlet/landuse_basis/gap/beach_walls_2lands'

# setup and save to shelf
# set up saving
save_file = 'py_save_file'

main_run, domain, mann_pts, wall_pts, points = rmw.loadmat(save_file, base_dir,
        grid_dir, save_dir, basis_dir)

pt_nos = range(points.shape[-1])

pa.nts_pcolor(main_run.nts_data, domain, points=pt_nos, path=save_dir)

示例#5

文件： concatenate_pair.py 项目： tmiesse/PolyADCIRC

#! /usr/bin/env python
# import necessary modules
import polyadcirc.run_framework.random_wall as rmw
import numpy as np

base_dir = '/h1/lgraham/workspace'
grid_dir = base_dir + '/ADCIRC_landuse/Inlet/inputs/poly_walls'
save_dir1 = base_dir + '/ADCIRC_landuse/Inlet/runs/poly_wall1'
save_dir2 = base_dir + '/ADCIRC_landuse/Inlet/runs/poly_wall2'
basis_dir = base_dir + '/ADCIRC_landuse/Inlet/landuse_basis/gap/beach_walls_2lands'

# setup and save to shelf
# set up saving
save_file = 'py_save_file'

main_run, domain, mann_pts1, wall_pts1, points1 = rmw.loadmat(
    save_file, base_dir, grid_dir, save_dir1, basis_dir)
other_run, domain, mann_pts2, wall_pts2, points2 = rmw.loadmat(
    save_file, base_dir, grid_dir, save_dir2, basis_dir)

cated = main_run.concatenate(other_run, points1, points2)

mdat = dict()
mdat['points'] = cated[1]
mdat['mann_pts'] = np.concatenate((mann_pts1, mann_pts2),
                                  axis=points1.ndim - 1)
mdat['wall_pts'] = np.concatenate((wall_pts1, wall_pts2),
                                  axis=points1.ndim - 1)

main_run.update_mdict(mdat)
main_run.save(mdat, 'cat_file')